筛选具有平均均方根值的DataFrame

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的列。