Python:转换数据集

Question

我有以下数据集

            id  type  value
0             1   A    10
1             1   C   120
2             2   B    20
3             2   C    40
4             3   A    10
5             3   B    50

我想在 python 中将其转换为 (1,A,10,C,120) (2,B,20,C,40) (3,A,10,B,50)

任何建议将不胜感激

Answer 1

执行 groupby w.r.t id 列。通过将其他两列转换为 list 并在旁边添加与每个组编号的 id 相对应的唯一值来遍历每个组。最后，将它们转换为 tuple 并将它们附加到列表中。

grouped = df.groupby('id')
L = []
for _, grp in grouped:
    L.append(tuple(grouped.get_group(_)['id'].unique().tolist() + grp[['type','value']].values.ravel().tolist()))
print(L)
#[(1, 'A', 10, 'C', 120), (2, 'B', 20, 'C', 40), (3, 'A', 10, 'B', 50)]