python - 在 Pandas 中，选择 "where"时，删除未满足条件而不是填充NAN的列

Question

在 Numpy 中，使用 where 将为您提供原始数组的子集。例如，

import numpy as np
np.where(np.arange(5)>2)[0]

将返回 array([3, 4])。我想对 Pandas 做类似的事情。但是，如果我像这样定义一个类似的 DataFrame:

import pandas as pd
df = pd.DataFrame(data=range(5)).T

结果 df 看起来像

   0  1  2  3  4
0  0  1  2  3  4

并申请

df.where(lambda x: x>2)

我明白了

    0   1   2  3  4
0 NaN NaN NaN  3  4

但是，我想得到这个:

    3  4
0   3  4

省略不满足条件的列。如何做到这一点？

Answer 1

IIUC 您可以调用 dropna 传递 axis=1 以删除包含任何 NaN 值的列:

In [272]:
df[df > 2].dropna(axis=1)

Out[272]:
   3  4
0  3  4