如何在pandas.DataFrame.query()中使用负数表达式？

这个报错是因为pandas的表达式解析器在处理ColumnName<(-1000)这种带括号的负数写法时，会错误地将(-1000)识别为一元操作符对象，从而抛出AttributeError: 'UnaryOp' object has no attribute 'value'。下面给你几个可行的解决方案：

方案1：去掉负数的括号，直接写表达式 #

把表达式改成不带括号的形式，pandas就能正确解析负数了：

expression_string = 'ColumnName < -1000'
output_dataframe = dataframe.query(expression_string)

方案2：调换比较两边的位置 #

把负数放在比较运算符的左侧，同样可以避免解析问题：

expression_string = '-1000 > ColumnName'
output_dataframe = dataframe.query(expression_string)

方案3：用变量引用负数（推荐） #

如果需要复用这个阈值或者让代码更清晰，可以把负数定义成变量，然后用@符号在query表达式中引用：

threshold = -1000
output_dataframe = dataframe.query('ColumnName < @threshold')

测试示例 #

你可以用下面的代码验证这些方案的有效性：

import pandas as pd

# 创建测试DataFrame
dataframe = pd.DataFrame({'ColumnName': [1000, -2000, 500, -1500]})

# 方案1测试
expr1 = 'ColumnName < -1000'
result1 = dataframe.query(expr1)
print("方案1结果：")
print(result1)

# 方案2测试
expr2 = '-1000 > ColumnName'
result2 = dataframe.query(expr2)
print("\n方案2结果：")
print(result2)

# 方案3测试
threshold = -1000
result3 = dataframe.query('ColumnName < @threshold')
print("\n方案3结果：")
print(result3)

三个方案都会返回包含-2000和-1500的行，完全符合预期。

内容的提问来源于stack exchange，提问作者Mridula Gunturi