python - 关于 pandas 按功能分组的问题？

Question

我有以下数据集:

productid att
12        block10
12        block20
12        clean
12        screw
12        nail
13        hard
13        cover
14        round
14        narrow
14        black
15        block4

我想根据产品 ID 对数据框进行分组并得到以下结果:

productid att
12        block10 block20 clean screw nail
13        hard cover
14        round narrow black
15        block4

我可以使用 pandas.groupby('productid') 对数据进行分组，但我不确定如何将特定 Productid 的所有数据写入以空格分隔的单行。

Answer 1

groupby 在“productid”上，然后应用联接:

In [6]:
df.groupby('productid')['att'].apply(' '.join)

Out[6]:
productid
12    block10 block20 clean screw nail
13                          hard cover
14                  round narrow black
15                              block4
Name: att, dtype: object