如何按索引和列分组

2024-03-29 12:53:36 发布

您现在位置:Python中文网/ 问答频道 /正文

假设我有一个简单的dataframe,它有一个datetime索引和三列—一列是值,一列是聚合指示符,最后一列是唯一标识符。索引在整个数据帧中不是唯一的,因为它在指示器的多个实例上是重复的。ie看起来是这样的:

index, val, aggregating indicator, unique_ref
1-Jan, 1, set_a,r1
2-Jan, 2, set_a,r2
1-Jan, 3, set_a,r3
2-Jan, 4, set_a,r4
1-Jan, 5, set_b,r5
2-Jan, 6, set_b,r6

是否可以聚合(在一个步骤中)val并按索引和聚合指示符分组?你知道吗

我想要的结果是

index, sum, aggregating_indicator
1-Jan, 4, set_a // ie 1+3
2-Jan, 6, set_a // ie 2+4
1-Jan, 5, set_b
2-Jan, 6, set_b

分两步做就可以了。。。但我想知道是否有可能用一种简洁的方式来做。你知道吗


Tags: 数据实例refdataframedatetimeindexval标识符
1条回答
网友
1楼 · 发布于 2024-03-29 12:53:36

你可以这样做:

>>> df.reset_index().groupby(['index','aggregating indicator'])['val'].sum().reset_index()
   index aggregating indicator  val
0  1-Jan                 set_a    4
1  1-Jan                 set_b    5
2  2-Jan                 set_a    6
3  2-Jan                 set_b    6

相关问题 更多 >