Pandas.query，定义变量

Question

在这个例子中，我想从dataframe中选择datevalue大于7440的所有值。但是在下面的代码中它不起作用，因为我收到“dv is not defined”。那么我如何在熊猫数据框中定义它呢。任何帮助都会很好。在

dv = 7440
values = df.query('datevalue > dv' and country = "usa"')

Answer 1

为变量添加@，并将=更改为双==以实现比较相等：

df = pd.DataFrame({'country':['usa','canada','slovakia','nepal'],
                   'datevalue':[7800,8440,10000,50000]})

print (df)
    country  datevalue
0       usa       7800
1    canada       8440
2  slovakia      10000
3     nepal      50000

dv = 7440
values = df.query('datevalue > @dv and country == "usa"')
print (values)
  country  datevalue
0     usa       7800