Попытка сложить ряды с пандами несколько сложным способом

Question

У меня есть DataFrame:

df = np.DataFrame = {'Year' : [2010, 2011, 2012, 2013, 1922, 1923, 1924, 1925],  
                     'ID' : ['A', 'A', 'A', 'A', 'B', 'B', 'B'],
                     'Data1' : [1, 2, 3, 4, 2, 3, 4], 
                     'Data2' : [2, 2, 2, 2, 3, 3, 3]}

       df
Year ID Data1 Data2
2010  A   1     2
2011  A   2     2
2012  A   3     2
2013  A   4     2
1922  B   2     3
1923  B   3     3
1924  B   4     3

Я хотел бы суммировать итоговые данные1 и данные2 для каждого идентификатора, чтобы каждый год давал сумму всех данных1 или данных2 за этот год.

В конечном счете, я хотел бы, чтобы мой DataFrame выглядел так:

    df
Year ID Data1 Data2
2010 A    1     2
2011 A    3     4
2012 A    6     6
2013 A    10    8
1922 B    2     3
1923 B    5     6
1924 B    9     9

У меня более 100 тыс. Строк, поэтому, если возможно, это нужно делать без итераций.

Answer 1

используйте assign и groupby

df = df.assign(**df.groupby('ID')['Data1', "Data2"].cumsum())
print(df)

    Year   ID  Data1    Data2
0   2010    A    1       2
1   2011    A    3       4
2   2012    A    6       6
3   2013    A    10      8
4   1922    B    2       3
5   1923    B    5       6
6   1924    B    9       9