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

Question

当我使用变量而不是值时，我在 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 是 undefined.But 已定义的消息。我不能使用硬编码值，因为我需要自动化并取决于 variable_name、select 相关行的值。

我是不是漏掉了什么？

谢谢

Answer 1

你应该使用 @variable_name 和 @

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