python - 如果一列中的至少值满足使用 Pandas any 的条件，则选择行

Question

我有以下数据框:

import pandas as pd
df = pd.DataFrame({ 'gene':["foo","bar","qux","woz"], 'cell1':[5,0,1,0], 'cell2':[12,90,13,0]})
df = df[["gene","cell1","cell2"]]

看起来像这样:

  gene  cell1  cell2
0  foo      5     12
1  bar      0     90
2  qux      1     13
3  woz      0      0

我想做的是根据从第 2 列开始的至少一个值是否大于 2 来选择行。产生这个:

  gene  cell1  cell2
0  foo      5     12
1  bar      0     90
2  qux      1     13

我试过了，但它没有给我想要的东西:

df[(df.values > 2).any(axis=1)]

正确的做法是什么？

Answer 1

您应该选择 cell1 和 cell2，然后将它们与 2 进行比较。示例 -

In [4]: df[(df[['cell1','cell2']] > 2).any(axis=1)]
Out[4]:
   cell1  cell2 gene
0      5     12  foo
1      0     90  bar
2      1     13  qux