下面是一个示例代码,用于比较两列并添加条件:
import pandas as pd
# 创建示例数据
data = {
'Column1': [1, 2, 3, 4, 5],
'Column2': [2, 4, 6, 8, 10]
}
df = pd.DataFrame(data)
# 添加条件列
df['Condition'] = ''
# 比较两列并添加条件
for index, row in df.iterrows():
if row['Column1'] > row['Column2']:
df.at[index, 'Condition'] = 'Column1 > Column2'
elif row['Column1'] < row['Column2']:
df.at[index, 'Condition'] = 'Column1 < Column2'
else:
df.at[index, 'Condition'] = 'Column1 = Column2'
# 打印结果
print(df)
输出结果如下:
Column1 Column2 Condition
0 1 2 Column1 < Column2
1 2 4 Column1 < Column2
2 3 6 Column1 < Column2
3 4 8 Column1 < Column2
4 5 10 Column1 < Column2
在这个示例中,我们创建了一个包含两列数据的DataFrame,并添加了一个名为“Condition”的空条件列。然后,通过使用iterrows()
方法遍历DataFrame的每一行,比较Column1
和Column2
的值,并根据比较结果将相应的条件值分配给Condition
列。
请注意,这只是一个简单的示例,实际应用中可能需要根据实际需求进行修改。
上一篇:比较两列并生成新列
下一篇:比较两列并突出显示单元格