python - 删除多值列

Question

我有一个数据框。

   A  B
0  2  3
1  2  4
2  3  5

如果一列的值有超过 2 个不同的值，我将删除。期待输出:

   A
0  2
1  2
2  3

Answer 1

你可以使用.nunique()和.loc，传递一个 bool 值

df = pd.DataFrame({'A': {0: 2, 1: 2, 2: 3}, 'B': {0: 3, 1: 4, 2: 5}})
df.loc[:, (df.nunique() <= 2)]

   A
0  2
1  2
2  3

另一种方法(归功于 this answer ):

criteria = df.nunique() <= 2
df[criteria.index[criteria]]