用平均树hold过滤数据帧

Question

我有一个数据帧，我只想保留列，当它们的平均值超过某个treshhold时

我的代码如下所示：

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\u ONLY\u COLUMNS”这样的函数可以将它简化为一行

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

Answer 1

在此处使用^{}：

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

这将自动保留条件为真的列