python - 将 pandas 中的行值扩展为多列

Question

我有一个包含两列的数据框:id 和vendor_name，如下所示:

 id   vendor_name
  1   vendor1
  1   vendor2
  2   vendor1

我想将其转换如下:

 id   vendor_name1 vendor_name2
  1    vendor1        vendor2  
  2    vendor1        nan

有什么办法可以在 pandas 中实现这一点吗？

Answer 1

使用Series.unstack用于使用 GroupBy.cumcount 创建的新列名称进行 reshape :

df1 = (df.set_index(['id', df.groupby('id').cumcount().add(1)])['vendor_name']
        .unstack()
        .add_prefix('vendor_name')
        .reset_index())
print (df1)
   id vendor_name1 vendor_name2
0   1      vendor1      vendor2
1   2      vendor1          NaN