python - 如何比较同时包含字符串和整数的列？ python Pandas

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