python - 如何根据 Pandas 中的真/假条件选择列

Question

我想选择满足以下条件的列。

mask = df.select_dtypes(include = ['object']).apply(pd.Series.nunique) > 100

rbc      False
pc       False
pcc      False
ba       False
htn      False
dm       False
cad      False
appet    False
pe       False
ane      False
class    False
lev_a     True
lev_b     True
l_x      False
dtype: bool

Answer 1

问题很可能是掩码中的行数与原始数据框中的列数不同(假设您的列是对象和非对象的混合体)。

根据您的设置，这应该会返回满足您的标准的列(然后您可以将其用作 loc[] 的输入):

mask.index[mask]

你想在这里说 mask.index 而不是 mask.columns 因为你的掩码是一个 pandas 系列，其中索引由原始数据帧中的列组成 df.

这是一个更完整的示例数据示例(在您以后的问题中，请提供您自己的示例数据，您通常会得到更多更好的答案):

df = pd.DataFrame( np.random.choice(list('abcde'),[10,5]), 
                          columns=list('ABCDE'))

mask = df.select_dtypes(include=['object']).apply(pd.Series.nunique) > 4

mask.index[mask]

结果:

Index(['B', 'C'], dtype='object')