如何在Pandas中使用查询内的变量?

时间:2015-05-20 04:28:02

标签: python variables indexing pandas

当我使用变量而不是值时,我在panda中查询数据框时遇到了问题。

df2 = pd.read_csv('my.csv')
query=df2.query('cc_vehicle_line==7')

工作正常,但

df2 = pd.read_csv('my.csv')
query=df2.query('cc_vehicle_line==variable_name')

它抛出了variable_name未定义的消息。但是它被定义了。我不能使用硬编码值,因为我需要自动化并根据variable_name的值,选择相关的行。

我错过了什么吗?

由于

1 个答案:

答案 0 :(得分:17)

您应该将@variable_name@

一起使用
query=df2.query('cc_vehicle_line==@variable_name')