如何正确使用pandas中的变量进行分组？

Question

我想向用户请求FSA：

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

然后我想在FSA中显示所有的行走量。如果我直接在参数（M9A）中放入我正在查找的FSA，它工作得很好：

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

但是，如果我尝试使用FSA\u输入变量，如下所示，它将不起作用：

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

在这种情况下，使用变量的正确方法是什么？你知道吗

Answer 1

我认为问题在于代码仍然将FSA_input识别为空字符串。 尝试更改：

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