pandas.core.groupby.GroupBy.sum

Compute sum of group values.

return

Series or DataFrame Computed sum of values within each group.

Limitations

• This function may reveal slower performance than Pandas* on user system. Users should exercise a tradeoff between staying in JIT-region with that function or going back to interpreter mode.

Examples

Compute sum of groups, excluding missing values.

import pandas as pd
from numba import njit


@njit
def df_groupby_sum():
    df = pd.DataFrame({'A': [1, 2, 3, 1, 2, 3, 3, 3, 2],
                       'B': [0, 1, 5, 0, 2, 4, 3, 2, 3],
                       'C': [1, 2, 3, 4, 5, 6, 7, 8, 9]})
    out_df = df.groupby('A').sum()

    # Expect DataFrame of
    # {'B': [0, 6, 14], 'C': [5, 16, 24} with index=[1, 2, 3]
    return out_df


print(df_groupby_sum())

$ python ./dataframe/groupby/dataframe_groupby_sum.py
      B     C
1   0.0   5.0
2   6.0  16.0
3  14.0  24.0