pandas 数据帧多行查询

问题描述

假设我有一个数据帧

import numpy as np
import pandas as pd

df = pd.DataFrame(np.random.randint(10, size=(10,3)), columns=['a', 'b', 'c'])

如果我现在尝试使用query方法查询它：

这行得通：

df.query('''a > 3 and b < 9''')

这将引发错误：

df.query(
    '''
        a > 3 and
        b < 9
    '''
)

我尝试了多行字符串的多种变体，但结果总是出现以下错误：

~/ven/lib/python3.6/site-packages/pandas/core/computation/eval.py in eval(expr, parser, engine, truediv, local_dict, global_dict, resolvers, level, target, inplace)
    306     if multi_line and target is None:
    307         raise ValueError(
--> 308             "multi-line expressions are only valid in the "
    309             "context of data, use DataFrame.eval"
    310         )

ValueError: multi-line expressions are only valid in the context of data, use DataFrame.eval

有人知道怎么使它工作吗？ 问题是，在现实中，我有一个很长的查询要做，必须在一行中写下所有内容，这将是非常不方便的。 我知道我可以改用布尔索引，但我的问题只是关于如何将multiline与query方法一起使用。

谢谢

---

解决方案

使用多行字符反斜杠()

示例：

df = pd.DataFrame(np.random.randint(10, size=(10,3)), columns=['a', 'b', 'c'])
print(df.query(
    '''
        a > 3 and 
        b < 9
    '''
))