Pandas DataFrame中的哪些列是二進制的？

Question

我有一個大量的列的熊貓數據框，我需要找到哪些列是二進制的（只有值0或1），而不看數據。應該使用哪個函數？

Answer 1

據我所知，沒有直接的功能來測試這個。相反，您需要根據數據的編碼方式（例如1/0，T/F，True/False等）構建一些東西。另外，如果你的列有一個缺失值，整個列將被編碼爲一個float而不是int。

在下面的例子中，我測試所有唯一的非空值是'1'還是'0'。它返回所有這些列的列表。

df = pd.DataFrame({'bool': [1, 0, 1, None], 
        'floats': [1.2, 3.1, 4.4, 5.5], 
        'ints': [1, 2, 3, 4], 
        'str': ['a', 'b', 'c', 'd']}) 

bool_cols = [col for col in df 
      if df[[col]].dropna().unique().isin([0, 1]).all().values] 

>>> bool_cols 
['bool'] 

>>> df[bool_cols] 
    bool 
0  1 
1  0 
2  1 
3 NaN 

Answer 2

要在答案擴大正上方，用value_counts（）指數，而不是唯一的（）應該做的伎倆：

bool_cols = [col for col in df if 
       df[col].dropna().value_counts().index.isin([0,1]).all()]