python - Pandas Dataframe 转置到不同数量的列

Question

我希望转置如下内容:

    ID    name
1     1    name
2     1    name1
3     2    name2
4     2    name3
5     2    name4

变成这样的东西:

     ID    col1  col2  col3
1     1    name  name1 Null
2     2    name2 name3 name4

Answer 1

您可以尝试以下操作，首先创建一个列变量，然后使用 pivot:

df['col'] = 'col' + (df.groupby('ID').cumcount() + 1).astype(str)
df.pivot('ID', 'col', 'name')

#col    col1     col2    col3
#ID         
#1      name    name1    None
#2     name2    name3   name4