pandas - 如何将值存储在变量中并使用该变量过滤熊猫中的数据

Question

我必须接受我存储在变量中的输入并使用该值过滤掉这样的数据

u = 'Jun'
duration = compda.query("Month==u").groupby('name').duration.mean().reset_index().values.tolist()

但是，这不起作用

duration = compda.query("Month=='Jun'").groupby('name').duration.mean().reset_index().values.tolist()

作品

标签： pandaspandas-groupby

Answer 1

这应该可行，您可以使用 @ 传递变量（此处的文档）：

u = 'Jun'
duration = compda.query('Month == @u').groupby('name').duration.mean().reset_index().values.tolist()