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
    '''
))

相关文章