Блог

Я пытаюсь создать функцию, которая принимает входное имя значения в столбце, и это значение затем будет использоваться в функции df.query. Однако я не могу понять, как сделать ее переменной, которую она распознает как входные данные.

Это то, что у меня есть прямо сейчас:

 def gettingWeeks(stateAbbr, stateName):

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

Но он не распознает StateName . Есть ли способ сделать это?
Спасибо!

DataFrame.query Метод Pandas ожидает строку выражения, созданную в соответствии с ее конкретным синтаксисом. Чтобы использовать переменные из текущего пространства имен, вы должны использовать @ символ перед именем переменной:

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

Должно работать нормально.

Вот документ.