如何在Pandas中保留至少一列满足条件的行

时间:2015-01-13 01:52:00

标签: python pandas

我有以下DF:

In [1]: import pandas as pd

In [2]: mydict = {'foo':[0, 0.3,5], 'bar':[1,0.55,0.1], 'qux': [0.3,4.1,4]}

In [3]: df = pd.DataFrame.from_dict(mydict, orient='index')

In [4]: df
Out[4]:
       0     1    2
qux  0.3  4.10  4.0
foo  0.0  0.30  5.0
bar  1.0  0.55  0.1

我想要做的是保留行,如果至少有一列是> 2。 最终输出如下:

       0     1    2
qux  0.3  4.10  4.0
foo  0.0  0.30  5.0

在熊猫中做到这一点的方法是什么?

1 个答案:

答案 0 :(得分:8)

In [201]: df.loc[(df > 2).any(axis=1)]
Out[201]: 
       0    1  2
qux  0.3  4.1  4
foo  0.0  0.3  5