python - 通过分组分隔列中的数据

Question

这个问题已经有答案了:

How can I pivot a dataframe? (5 个回答)

已关闭 4 年前。

我有一个多列数据框，并且希望通过根据另一列对特定列中的数据进行分组来分隔数据。

这是一个例子:

ID     Name   Score
1      John   100
2      Lisa   80
3      David  75 
4      Lisa   92
5      John   89
6      Lisa   72

我希望我的输出是这样的:

index  John  Lisa  David
0      100   80    75
1      89    92    NaN
2      NaN   72    NaN

我知道我可以轻松使用代码:

df[df['Name'] == 'John]]并将约翰在数据框中的所有分数分开，但由于我的数据框非常大，我希望有一个干净的输出，就像我在这里显示的那样。

Answer 1

您可以使用pivot_table 。您可以生成按 Name 对数据帧进行透视分组所需的索引，并采用 cumcount :

ix = df.groupby('Name').cumcount()
pd.pivot_table(df,'Score', ix, 'Name')

Name  David   John  Lisa
1      75.0  100.0  80.0
2       NaN   89.0  92.0
3       NaN    NaN  72.0