仅返回满足 where 子句的数据框列

Question

从任意数据框开始，我想返回一个数据框，其中仅包含具有多个不同值的那些列。我有：X = df.nunique()喜欢：    Id               5    MSSubClass       3    MSZoning         1    LotFrontage      5    LotArea          5    Street           1    Alley            0    LotShape         2然后我将其从系列转换为数据框：X = X.to_frame(name = 'dcount')然后我使用 where 子句只返回大于 1 的值：X.where(X[['dcount']]>1)看起来像：                   dcount    Id                5.0    MSSubClass        3.0    MSZoning          NaN    LotFrontage       5.0    LotArea           5.0    Street            NaN    Alley             NaN    LotShape          2.0    ...但我现在只想要那些没有 dcount = 'NaN' 的 column_names（在 X 的索引中），以便我最终可以返回到我的原始数据帧 df 并将其定义为：df=df[[list_of_columns]]这应该怎么做？我尝试了十几种方法，这是一个 PitA。我怀疑有一种方法可以用 1 或 2 行代码来完成。

Answer 1

您可以使用布尔索引并避免将计数系列转换为数据帧：

counts = df.nunique()

df = df[counts[counts > 1].index]

关键是要注意您系列的索引counts是列标签。因此，您可以过滤系列，然后通过pd.Series.index.

这是一个演示：

df = pd.DataFrame({'A': [1, 1, 1], 'B': [1, 2, 3],

                   'C': [4, 5, 5], 'D': [0, 0, 0]})

counts = df.nunique()

df = df[counts[counts > 1].index]

print(df)

   B  C

0  1  4

1  2  5

2  3  5