在 DataFrame 中将布尔值转换为字符串

Question

源于这篇文章，但我发现解决方案有一个小问题。

使用df = df.replace({True: 'TRUE', False: 'FALSE'}), 如果一个值是1或0它将被替换。我只想替换实际True或False的值<class 'bool'>。

我怎样才能做到这一点？

Answer 1

您可以使用 来执行此操作df.where，因此您只需替换bool类型。

import pandas as pd

mask = df.applymap(type) != bool
d = {True: 'TRUE', False: 'FALSE'}

df = df.where(mask, df.replace(d))

输入：

df = pd.DataFrame({'vals': [1, 0, 1.0, 'True', True, False],
                   'ids': [78, 'kitten', True, False, 'orange', 9]})

输出：

    vals     ids
0      1      78
1      0  kitten
2      1    TRUE
3   True   FALSE
4   TRUE  orange
5  FALSE       9