python - 检查 pandas 列是否仅包含 0's or 1' s

Question

我想知道是否有办法检查 pandas 列是否只包含 0 或 1。这可以使用 df.groupby('col').count() 来完成，然后验证是否只有两个索引并检查 0 和 1 是否是索引的一部分。有更好的办法吗？

Answer 1

使用Series.isin通过 Series.all 测试所有值是否为 True :

它返回True或False标志。在这里查看它的实际效果。让我们考虑这个数据框:

df = pd.DataFrame({'col1':[0,1,0],
                   'col2':[2,3,1]})
print (df)
   col1  col2
0     0     2
1     1     3
2     0     1

test = df['col1'].isin([0,1]).all()
True

test = df['col2'].isin([0,1]).all()
False