如何比较熊猫数据框的两个不同列的值？

Question

我想比较 Pandas 数据帧的两个不同列的值并返回一个布尔列表或其他东西，我们可以使用它来绘制显示真假命题的饼图。

在下图中，我想比较 toss_winner 列和 winner 列。

https://i.stack.imgur.com/Z2TDE.png

我试过这样做：

<块引用>

df['toss_winner'].equals(df['winner'])

但它比较整个列。有人可以帮我吗？

Answer 1

您可以使用 == 运算符或 eq 方法

df['toss_winner'] == df['winner']

样品运行：

df = pd.DataFrame({"A": np.random.randint(0,10,100), "B": np.random.randint(0,10,100)})
df['A'] == df['B']
Out[159]: 
0     False
1     False
2     False
3     False
4     False
      ...  
95    False
96    False
97    False
98    False
99     True
Length: 100, dtype: bool

Answer 2

应该是这样：

df['flag'] = df['toss_winner']==df['winner']