如何在pandas groupby中正确使用变量？

Question

我想问用户FSA：

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

然后我想在FSA中显示所有Walk数量。如果我把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()