对于数据框df,我选择在“ a”或“ b”列中具有True值的行。
>>> df
Out[127]:
a b
0 False False
1 True True
2 True False
>>> con = (df['a'] == True) | (df['b'] == True)
>>> con
Out[129]:
0 False
1 True
2 True
dtype: bool
>>> df[con]
Out[130]:
a b
1 True True
2 True False
数据框中只有两列。对于实际代码,此类列的数量是一个变量。如何动态con
生成条件?
说,当df从a到z有26列时,我想要类似
>>> con = (df['a'] == True) | (df['b'] == True) | ... (df['y'] == True) | (df['z'] == True)
我可以用来获取所需的行
您可以使用DataFrame.any
:
>>> df = pd.DataFrame(np.random.choice([True]+[False]*5, size=(6,5)), columns=list("abcde"))
>>> df
a b c d e
0 False False False False False
1 False False True False False
2 False False True False False
3 False False False False False
4 False False False False True
5 False False False False False
>>> df.any(axis=1)
0 False
1 True
2 True
3 False
4 True
5 False
dtype: bool
>>> df[df.any(axis=1)]
a b c d e
1 False False True False False
2 False False True False False
4 False False False False True
和往常一样,df.loc[df.any(axis=1)]
如果您想确保原稿上有一个手柄,可以使用。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句