python - 使用 pandas 数据框对字符进行元素明智逻辑运算

Question

我正在使用一个数据框，其中每个条目都是代表不同强度级别的 A-E 字符

例如:

数据框

df=

        1   2   3
    0   C   C   A
    1   0   D   B
    2   E   0   A

我想将 A 到 E 不同强度级别的数据阈值化为二进制矩阵。因此，我想执行逐元素比较。

如果是数字，则很简单:df > 0但是，当我尝试使用字符 df > 'A' 时，会抛出错误 “TypeError:无法将 ['A'] 与 block 值进行比较”

我当然可以做一个“for”循环，但是有没有一行优雅的解决方案可以使用 pandas 中的数据帧对字符执行元素逻辑操作？

Answer 1

我会考虑以下更通用的方法:

In [238]: df.astype(str).applymap(ord).sub(ord('@')).replace(-16,0)
Out[238]:
   1  2  3
0  3  3  1
1  0  4  2
2  5  0  1

地点:

0 - 0
1 - A
2 - B
3 - C
...