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

Question

昨晚来到这里，提出了有关分组依据的问题，并找到了一个非常快速的解决方案!我现在正在尝试解决一个问题，尽管我可以返回一个数据帧，其中仅包含我的 df 中的多个“complaint_type”之一，但是，我的 df 按“complaint_type”分组

我尝试过使用普通的过滤器语句，但是它说我有一个关键错误。下面显示了我的 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()