如何比较具有字符串和整数的列？蟒蛇熊猫

Question

我有一个带有两列的 pandas DataFrame，一列同时包含整数（1-23）和字符串（X 或 Y），另一列仅包含数字。我想比较值是否相等。我尝试过：

np.where(np.equal(item from column a, item from column b), 1, 0)

但这不起作用，因为那里有字符串。由于 X 和 Y，我也无法将所有值转换为整数。有什么建议吗？

Answer 1

Pandas 支持类型比较。您可以将pd.Series.__eq__ 用作两个相同类型系列之间的常规比较。

df = pd.DataFrame({'col1': [1, 2, 'hello', 4.5, 'text', 6, 7, 'errr', 9, 'test'],
                   'col2': range(1, 11)})

df['compare'] = (df['col1'] == df['col2']).astype(int)

print(df)

    col1  col2  compare
0      1     1        1
1      2     2        1
2  hello     3        0
3    4.5     4        0
4   text     5        0
5      6     6        1
6      7     7        1
7   errr     8        0
8      9     9        1
9   test    10        0