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

Question

当我使用变量而不是值时,我在熊猫中查询数据框时遇到问题.

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的值,选择相关的行.

我错过了什么吗？

谢谢

Answer 1

您应该使用@variable_name与@

query=df2.query('cc_vehicle_line==@variable_name')