要比较数据框的每一行中的两个列,你可以使用apply函数来遍历每一行,并进行比较。
以下是一个示例代码,假设你有一个名为df的数据框,其中包含两列"A"和"B":
import pandas as pd
# 创建示例数据框
data = {'A': [1, 2, 3, 4, 5],
'B': [5, 4, 3, 2, 1]}
df = pd.DataFrame(data)
# 创建一个函数来比较两列的值
def compare_cols(row):
if row['A'] > row['B']:
return 'A > B'
elif row['A'] < row['B']:
return 'A < B'
else:
return 'A = B'
# 使用apply函数来应用比较函数到每一行
df['Comparison'] = df.apply(compare_cols, axis=1)
# 打印结果
print(df)
输出结果为:
A B Comparison
0 1 5 A < B
1 2 4 A < B
2 3 3 A = B
3 4 2 A > B
4 5 1 A > B
在这个示例中,我们创建了一个名为compare_cols的函数,它接受一个行作为参数,并根据"A"列和"B"列的值返回"A > B"、"A < B"或"A = B"。然后我们使用apply函数将这个函数应用到数据框的每一行,并将结果存储在新的"Comparison"列中。最后,我们打印整个数据框来查看结果。