如何在Python中基于Pandas DataFrame按规则拆分x与y并应用条件

首先构造你给出的DataFrame：

import pandas as pd

# 构造目标DataFrame
data = {'a': [4, 2], 'b': [6, 3], 'c': [8, 1]}
df = pd.DataFrame(data)

接下来通过遍历DataFrame的列名，就能实现每次排除一列作为y、剩余列作为x的循环拆分：

# 遍历每一列，依次将其设为y
for target_col in df.columns:
    # x = 移除当前列后的所有数据
    x = df.drop(columns=[target_col])
    # y = 当前选中的单列数据
    y = df[target_col]
    
    # 在这里添加你需要对x和y应用的条件/处理逻辑
    # 示例：打印当前拆分结果
    print(f"y为列[{target_col}]时：")
    print("x:\n", x)
    print("y:\n", y)
    print("-" * 20)

拆分结果完全匹配你的步骤： #

• 步骤1（y=df[a]）：

  x:
    b  c
  0  6  8
  1  3  1
  y:
   0    4
  1    2
  Name: a, dtype: int64
  

• 步骤2（y=df[b]）：

  x:
    a  c
  0  4  8
  1  2  1
  y:
   0    6
  1    3
  Name: b, dtype: int64
  

• 步骤3（y=df[c]）：

  x:
    a  b
  0  4  6
  1  2  3
  y:
   0    8
  1    1
  Name: c, dtype: int64
  

你只需把循环内的示例代码替换为实际需要执行的操作（比如模型训练、数据校验等）即可。

内容的提问来源于stack exchange，提问作者Coder