要遍历数据框中所有列的组合,可以使用嵌套循环的方法。以下是一个示例代码:
import pandas as pd
from itertools import combinations
# 创建一个示例数据框
data = {'Col1': [1, 2, 3], 'Col2': [4, 5, 6], 'Col3': [7, 8, 9]}
df = pd.DataFrame(data)
# 获取数据框所有列名的组合
columns = df.columns
column_combinations = list(combinations(columns, 2)) # 这里选择两列的组合,可以根据需要调整数字
# 遍历所有列名组合
for combination in column_combinations:
col1, col2 = combination
print(f"Combination: {col1} + {col2}")
# 在这里可以对每个列名组合进行操作,例如计算它们的和、差、乘积等等
# 示例操作:计算两列的和
df['Sum'] = df[col1] + df[col2]
print(df)
df = df.drop('Sum', axis=1) # 移除添加的临时列
在上述示例中,我们首先创建了一个包含三列的示例数据框。然后,使用combinations
函数从数据框的列名中获取所有两列的组合。接下来,使用for
循环遍历列名组合,并对每个组合进行操作。示例中对每个组合计算了两列的和,并将结果添加到数据框中。最后,通过drop
函数移除添加的临时列。
下一篇:遍历数据框中唯一值列表