Pandas pivot_table 为每个条目创建一个新列

Question

我有一个像这样的熊猫数据框

我想将其转换为以下内容（不使用任何循环！）：

任何想法如何做到这一点？

如果图像不显示：

我有一个包含 2 列的数据框：名称和爱好。 我有以下几行：

Anna drawing
Anna swimming
Anna skiing
Lisa running
Lisa singing
Tom drawing

我想把它转换成一个有 4 列的数据框：名字、爱好 1、爱好 2、爱好 3。并有以下几行：

Anna drawing swimming skiing
Lisa running singing NaN
Tom drawing NaN NaN

Answer 1

使用gruopby和reset_index为每个名称组中的每个爱好创建数字索引。 重塑（unstack）以获得矩阵格式。

df = pd.DataFrame({'Name':['Anna','Anna','Anna','Lisa','Lisa','Tom' ],
                  'Hobby':['drawing','swimming','skiing','running','singing','drawing']})

result = (df.groupby('Name')['Hobby']
            .apply(lambda x:x.reset_index(drop=True))
            .unstack()
            .rename(columns=lambda x:f'Hobby {x+1}'))

编辑：两个爱好的结合

df_merge = (df.merge(df, on='Name')
              .assign(Hobby_combi=lambda x:x.Hobby_x+'|'+x.Hobby_y)
              .loc[lambda x:x.Hobby_x!=x.Hobby_y]
              .groupby(['Name','Hobby_combi'])
              .size()
              .unstack()
              .reindex(df['Name'].unique())
              .fillna(0, downcast='infer'))

df_merge.sum()