Как сохранить значение в переменной и использовать эту переменную для фильтрации данных в 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()

работает

Answer 1

Это должно сработать, вы можете использовать @ для передачи переменных ( документация здесь ):

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