Using a Variable in a .query function

Question

I'm trying to create a function that takes the input name of a value in a column and that value will then be used in a df.query function. However, I cannot figure out how to make it a variable that it recognizes as the input.

This is what I have right now:

def gettingWeeks(stateAbbr, stateName):

    stateCases = cases.query('state == stateName')

But it does not recognize stateName. Is there a way to do this? Thanks!

I159 · Accepted Answer · 2020-11-28 19:17:22Z

3

Pandas DataFrame.query method expects an expression string created accordingly to its specific syntax. To use variables from the current name space you have to use @ symbol before the name of the variable:

stateCases = cases.query("state == @stateName")

Should work fine.

Here is the doc.

answered Nov 28, 2020 at 19:17

I159

31.4k32 gold badges102 silver badges138 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Using a Variable in a .query function

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related