Cách tính tổng các giá trị được nhóm theo hai cột trong gấu trúc

Tôi có một DataFrame như thế này:

df = pd.DataFrame({
    'Date': ['2017-1-1', '2017-1-1', '2017-1-2', '2017-1-2', '2017-1-3'],
    'Groups': ['one', 'one', 'one', 'two', 'two'],
    'data': range(1, 6)})

    Date      Groups     data  
0  2017-1-1    one       1
1  2017-1-1    one       2
2  2017-1-2    one       3
3  2017-1-2    two       4
4  2017-1-3    two       5

Làm cách nào tôi có thể tạo một DataFrame mới như thế này:

    Date       one     two 
0  2017-1-1    3        0
1  2017-1-2    3        4
2  2017-1-3    0        5

pivot_table đã được thực hiện cho điều này:

df.pivot_table(index='Date',columns='Groups',aggfunc=sum)

kết quả trong

         data
Groups    one  two
Date
2017-1-1  3.0  NaN
2017-1-2  3.0  4.0
2017-1-3  NaN  5.0

Cá nhân tôi thấy cách tiếp cận này dễ hiểu hơn nhiều, và chắc chắn là nhiều pythonic hơn là một hoạt động nhóm phức tạp. Sau đó, nếu bạn muốn định dạng được chỉ định, bạn có thể dọn dẹp nó:

df.fillna(0,inplace=True)
df.columns = df.columns.droplevel()
df.columns.name = None
df.reset_index(inplace=True)

mang đến cho bạn

       Date  one  two
0  2017-1-1  3.0  0.0
1  2017-1-2  3.0  4.0
2  2017-1-3  0.0  5.0

Ma thuật đen gấu trúc:

df = df.groupby(['Date', 'Groups']).sum().sum(
    level=['Date', 'Groups']).unstack('Groups').fillna(0).reset_index()

# Fix the column names
df.columns = ['Date', 'one', 'two']

Kết quả df:

       Date  one  two
0  2017-1-1  3.0  0.0
1  2017-1-2  3.0  4.0
2  2017-1-3  0.0  5.0