Как правильно использовать переменную в группе панд?

Question

Я хочу попросить пользователя указать FSA:

FSA_input = ''
print('Input FSA:')
input(FSA_input)

Затем я хочу отобразить все количества прогулок в FSA.Это работает нормально, если я поставлю FSA, я смотрю прямо в аргументе (M9A):

print(df_HHE[df_HHE['SOFSA'] == 'M9A'].groupby('Walk')['CCh'].sum())

Но если я пытаюсь использовать переменную FSA_input, как показано ниже, она не работает:

print(df_HHE[df_HHE['SOFSA'] == FSA_input].groupby('Walk')['CCh'].sum())

Как правильно использовать переменную в этом случае?

Answer 1

Я думаю, что проблема в том, что код все еще распознает FSA_input как пустую строку.Попробуйте изменить:

FSA_input = ''
print('Input FSA:')
FSA_input = input()