要比较同一个pandas数据框中的两列值,并根据比较结果返回第三列的值,可以使用pandas的条件筛选功能。
下面是一个示例代码:
import pandas as pd
# 创建一个示例数据框
data = {'A': [1, 2, 3, 4, 5],
'B': [3, 2, 1, 4, 5],
'C': [10, 20, 30, 40, 50]}
df = pd.DataFrame(data)
# 比较两列的值,并根据比较结果返回第三列的值
df['D'] = df['C'].where(df['A'] > df['B'], df['C'])
print(df)
输出结果如下:
A B C D
0 1 3 10 10
1 2 2 20 20
2 3 1 30 30
3 4 4 40 40
4 5 5 50 50
在上面的示例中,我们比较了数据框中的两列A和B的值,如果A的值大于B的值,则返回第三列C的值,否则返回NaN。最后将结果存储在新列D中。