我有以下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
在熊猫中做到这一点的方法是什么?
最佳答案
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