python - 我正在使用 Pandas 从数据框中提取两列，但其中一列成为索引，然后在尝试访问该列时出现关键错误

Question

df.columns = df.columns.str.strip() ##found a fix for leading whitespaces
arrest_only_Y= df.loc[df['ARREST'] == 'Y']
arrest_only_Y_two_col=arrest_only_Y[["ARREST",'LOCATION DESCRIPTION','CASE#']]##running fine here 
arrest_only_Y_two_col.reset_index()
arrest_only_Y_two_col_groupby = arrest_only_Y_two_col.groupby('LOCATION DESCRIPTION').count() ##and here as well ## arrest_only_Y_two_col_groupby_desc=arrest_only_Y_two_col_groupby.sort_values(['ARREST'],ascending = False).head()
arrest_only_Y_two_col_groupby_desc.reset_index(drop = True)
arrest_only_Y_two_col_groupby_desc

在输出中，LOCATION DESCRIPTION 变成索引，我不能将它用作运行此代码的列

 locdesc_list = arrest_only_Y_two_col_groupby_desc['LOCATION 
 DESCRIPTION'].tolist()

我得到:Key Error : 'LOCATION DESCRIPTION'

Answer 1

替换你的行:

arrest_only_Y_two_col_groupby_desc.reset_index(drop=True)

与:

arrest_only_Y_two_col_groupby_desc.reset_index(inplace=True)