在python中对数据帧使用GROUPBY之后，如何过滤数据帧以返回特定类型的记录？

Question

昨晚带着一个关于group by的问题来到这里，找到了一个很快的解决方案！我现在正试图解决一个问题，虽然我可以返回一个数据帧，其中只包含df中的多个“投诉类型”中的一个，但是，我的df是按“投诉类型”分组的

我试着只使用一个普通的filter语句，但是它说我有一个键错误。下面显示了我的df以及我使用的语句

# Original Data Frame

NY311_df

# Creating a new df from the original based on group by

b_ct_df = NY311_df.groupby(['borough','complaint_type']).count()

#Trying to return the same df, but only where the complaint type is 
#'HEAT/HOT WATER'

b_ct_df[b_ct_df['complaint_type']=='HEAT/HOT WATER']

#The line above returns the error 'complaint_type' KeyError

我想退回每个行政区的“热水/热水”类型。我希望它看起来像布朗克斯：热水：456，布鲁克林：热水：543，等等

就在一个迷失的地方，是什么导致了这里的错误。干杯

Answer 1

备选方案要评论解决方案，您可以执行以下操作：

b_ct_df = NY311_df.groupby(['borough','complaint_type']).count().reset_index()
b_ct_df[b_ct_df['complaint_type']=='HEAT/HOT WATER']

或者你可以：

b_ct_df = NY311_df[NY311_df['complaint_type']=='HEAT/HOT WATER'].groupby(['borough','complaint_type']).count()