python - 拆分数据框并求和 [pandas]

Question

我有以下数据框(虚拟数据):

            score   GDP
country     
Bangladesh  6      12
Bolivia     4      10
Nigeria     3      9
Pakistan    2      3
Ghana       1      3
India       1      3
Algeria     1      3

我想根据 GDP 将其分成两组，然后将每组的分数相加。在GDP小于9的情况下:

           sum_score
country     
rich       13      
poor        5     

Answer 1

您可以使用 np.where 来创建您的 rich 和 poor 类别，然后 groupby 该类别和求和:

df['country_cat'] = np.where(df.GDP < 9, 'poor', 'rich')
df.groupby('country_cat')['score'].sum()

country_cat
poor     5
rich    13

您也可以通过不为类别创建额外的列来一步完成相同的操作(但在我看来，代码的可读性会降低):

df.groupby(np.where(df.GDP < 9, 'poor', 'rich'))['score'].sum()