python - 使用平均阈值过滤数据帧

Question

我有一个 DataFrame，当列的平均值超过某个阈值时，我只想保留列。

我的代码如下所示:

import pandas as pd
df =  pd.DataFrame(np.random.random((20,20)))
mean_keep= (df.mean() > 0.5)
mean_keep= mean_keep[mean_keep == True]
df_new = df[mean_keep.index]

并且它正在工作。但是我想知道是否有一个像“TAKE_ONLY_COLUMNS”这样的函数可以将其减少到一行，例如

df_new = df[TAKE_ONLY_COLUMNS(df.mean() > 0.5)]

Answer 1

使用df.loc[]这里:

df_new=df.loc[:,df.mean() > 0.5]
print(df_new)

这将自动保留条件为 True 的列。