python - reshape 具有不同长度行的数据框

Question

我在 python 中有一个数据框，如下所示:

ID    Value
001   0.5
001   0.2
001   0.5
001   0.0
002   0.4
002   0.6
002   0.6

我希望将数据 reshape 为如下形式:

ID    Val1    Val2   Val3   Val4
001   0.5     0.2    0.5    0.0
002   0.4     0.6    0.6    NaN

有人可以帮忙吗？我的第一个想法是使用“pivot”对数据进行去熔化，但没有表示“Val”位置的值，它无法按预期工作。

谢谢!

Answer 1

Grouppby 你的 ID 然后重置索引以保持列一致并取消堆叠

df.groupby('ID')['Value'].apply(lambda df: df.reset_index(drop=True)).unstack()
      0    1    2    3
ID
1   0.5  0.2  0.5  0.0
2   0.4  0.6  0.6  NaN

或者不使用 ID 作为索引:

df.sort_values('ID').groupby('ID')['Value'].apply(lambda df: df.reset_index(drop=True)).unstack().reset_index()
  ID    0    1    2    3
0  1  0.5  0.2  0.5  0.0
1  2  0.4  0.6  0.6  NaN