时间戳

首页 > 解决方案 > 按两列分组并得到总和？

python - 按两列分组并得到总和？

问题描述

x1 = [{'id1': 'Africa', 'id2': 'Europe', 'v': 1}, 
      {'id1': 'Europe', 'id2': 'North America', 'v': 5},
      {'id1': 'North America', 'id2': 'Asia', 'v': 2,}, 
      {'id1': 'North America', 'id2': 'Asia', 'v': 3}]

df = pd.DataFrame(x1)

我将如何按大洲分组并根据“v”列获得总和？

例如，我希望得到每个大陆的值总和，如下所示：

Africa: 1 (1)
Europe: 6 (1 + 5)
North America: 10 (5 + 2 + 3)
Europe: 6 (1 + 5)

标签： pythonpython-3.xpandasdata-manipulationpython-applymap

解决方案

使用melt和聚合sum：

s = df.melt('v').groupby('value')['v'].sum()
print (s)
value
Africa            1
Asia              5
Europe            6
North America    10
Name: v, dtype: int64

对于DataFrame：

df = df.melt('v', value_name='a').groupby('a', as_index=False)['v'].sum()
print (df)
               a   v
0         Africa   1
1           Asia   5
2         Europe   6
3  North America  10

推荐阅读