python - 如何将一些行转换为相同的索引 - Python

Question

我有一个表需要按以下方式进行转换

    ID      Item_ID  Rank
0   12      1068     1.0
1   12      8914     2.0
2   12      7997     3.0
3   13      311      4.0
4   13      123      5.0
5   13      39       6.0

到

ID    Item_ID_1  Item_ID_2  Item_ID_3
12    1068       8914       7997      
13    311        123        39

我已经尝试过pd.pivot，但给了我错误ValueError:索引包含重复的条目，无法 reshape ，但是，当然我有重复的条目。

请大家帮忙!

谢谢

Answer 1

使用groupby().cumcount()与 pivot_table() :

final=(df.assign(k=(df.groupby('ID').cumcount()+1).astype(str))
.pivot_table(index='ID',columns='k',values='Item_ID').add_prefix('Item_ID_')
 .reset_index().rename_axis(None,axis=1))

---

   ID  Item_ID_1  Item_ID_2  Item_ID_3
0  12       1068       8914       7997
1  13        311        123         39