Python Pandas Dataframe 添加基于重复的额外列 [重复]

Question

这个问题在这里已经有了答案：

我怎么能 pivot 一个 dataframe？ (3 个回答)

54 分钟前关闭。

今天我一直致力于合并和编辑数据框，但我一直被一个非常具体的部分所困扰。 我有一列包含某些水果的名称，然后是这样的人的名称：

水果	人
香蕉	杰克
香蕉	保罗
萝卜	南希
萝卜	悉尼
萝卜	简

请注意，“人员”列将始终是唯一的。 我的目标是用第三或第四列得到这样的东西：

水果	人1	人2	人3
香蕉	杰克	保罗
萝卜	南希	悉尼	简

我试过这个：

first = df.drop_duplicates(subset=['Zone','District','Area'],keep='First',inplace = True)
second = df.drop_duplicates(subset=['Zone','District','Area'],keep='Last',inplace = True)

并稍后将它们合并在一起，但这不会得到中间的那些，例如我的示例中的“Syndey”。 我发现了这个： 这里

但我不太了解groupby部分。 我希望这会有所帮助，并感谢您的时间和耐心。

Answer 1

使用df.groupby 、 Groupby.agg和pd.concat ：

In [890]: df1 = df.groupby('Fruit').agg(list).reset_index()
In [905]: res = pd.concat([df1.Fruit, pd.DataFrame(df1.Person.tolist(), index=df1.index)], 1)

In [906]: res
Out[906]: 
    Fruit      0       1     2
0  Banana   Jake    Paul  None
1  Carrot  Nancy  Sydney  Jane

Answer 2

df = df.assign(_=lambda x: 'Person'+(x.groupby('Fruit').transform(
    'cumcount')+1).astype(str)).pivot(columns='_', index='Fruit')
print(df)

结果：

        Person                
_      Person1 Person2 Person3
Fruit                         
Banana    Jake    Paul     NaN
Carrot   Nancy  Sydney    Jane