Суммирование и группировка по нескольким индексам pandas

Question

У меня есть фрейм данных с годами в качестве индексов:

index  deaths_pop

2010      2
2011      6
2012      2
2013      8
2014      10
2015      4
2016      2

Я хочу сгруппировать и суммировать по индексу следующим образом:

index    deaths_pop

=<2012     10
>2012      24

Как я могу это сделать?

Answer 1

Попробуйте groupby на pd.cut

(df.groupby(pd.cut(df.index, [0, 2012, np.inf], labels=['=<2012', '>2012']))
   .deaths_pop.sum())

Out[849]:
=<2012    10
>2012     24
Name: deaths_pop, dtype: int64