pandas 如何在groupby之后以下面的格式打印 Dataframe ?

qlzsbp2j  于 2023-01-28  发布在  其他
关注(0)|答案(1)|浏览(154)

我有一个python格式的 Dataframe ,格式如下

我试着创建一个groupby命令,输出如下:

但是,我希望输出在下面的格式:

我怎样才能在python(jupyter notebook)中得到上述格式的输出?

rsaldnfx

rsaldnfx1#

下面是一个方法:

import pandas as pd

df  = pd.DataFrame(
    [
        [1, 'John', 'Milk', 5],
        [2, 'John', 'Pencil', 3],
        [3, 'John', 'Bread', 2],
        [4, 'Victor', 'Milk', 1],
        [5, 'Sarah', 'Pencil', 4],
        [6, 'Sarah', 'Bread', 2],
    ],
    columns=['ID', 'Name', 'Product', 'Quantity']
)

groupby_prod = dict(df.groupby('Product', as_index=False).agg({'ID': lambda x: list(x)}).values)
print(groupby_prod)
# {'Bread': [3, 6], 'Milk': [1, 4], 'Pencil': [2, 5]}

相关问题