Python创建数字序列并按组追加

Question

我想按组 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...]

有人有简单的指导吗？

Answer 1

您可以重复索引 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]