通过查询选择所有列均满足条件的位置

Question

假设我在Pandas中有一个由几列组成的数据框。 我想选择数据框的子集，其中所有列都在两个值min和max 。

如何使用query做到这一点？ 我是否需要在表达式中一一指定每个列名，例如：

df.query('(A<{max} & A>{min}) & (B{min} & B{max}) & (C{min} & C{max})'.format(min,max))

除了查询之外，我还有什么其他选择？ 也许使用直接满足条件的索引？

Answer 1

对于您的用例，我将以非查询方式进行操作，构造一个布尔框架，然后调用.all方法：

>>> minval, maxval = 20, 80
>>> df = pd.DataFrame(np.random.randint(0, 100, (20,6)))
>>> df[((df > minval) & (df < maxval)).all(axis=1)]
     0   1   2   3   4   5
2   74  30  30  76  31  66
10  49  39  71  43  30  50

我认为手动构造扩展query的优势并不值得。

Answer 2

我不确定为什么您认为需要在这里query 。 但是，如果您想这样做，就可以。 您已经在使用该format语句动态建立查询字符串了； 您只需要通过加入理解就可以使其更具动态性。 例如：

qs = ' & '.join('{col}<{max} & {col}>{min}'.format(col, min, max)
                for col in df.columns)
df.query(qs)

我个人将以@DSM建议的方式执行此操作。 除非出于某种原因实际上需要查询字符串（也许是将其打印出来？），否则构建字符串以将其评估为表达式而不是构建表达式通常是错误的答案。