我想按组 ID 附加一系列数字。
我有一个如下所示的专栏:
ID [123, 124...]
我需要将数字序列 (1:485) 附加到每个唯一 ID，以便最终数据框如下所示:
ID [123, 123, 123, 123, 124, 124, 124, 124...]
Number [1, 2, 3, 4, 1, 2 , 3, 4...]
有人有简单的指导吗？
最佳答案
您可以重复索引 485 次并使用它来扩展 loc。 cumcount 然后给出每个 ID 的 1...485 计数:
new_df = df.loc[df.index.repeat(485)].reset_index(drop=True)
new_df["Number"] = new_df.groupby("ID").cumcount().add(1)
获取
>>> new_df
ID Number
0 123 1
1 123 2
2 123 3
3 123 4
4 123 5
.. ... ...
965 124 481
966 124 482
967 124 483
968 124 484
969 124 485
[970 rows x 2 columns]
关于Python 创建数字序列并按组附加，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/68499949/