如何為每個唯一列值 pandas dataframe 添加行系列？

Question

我想對我的數據框中的行進行編號（並將其添加為一列），其中每個不同的Number再次從 1 開始計數。 我試過使用df['Row number'] = np.arange(len(df))但這給出了行的連續編號。

我擁有的數據框示例：

Number Value
1234   a
1234   b
1234   x
5678   t
5678   y
5678   p

我想要的數據框示例：

Number Value   Row number
1234   a       1
1234   b       2
1234   x       3
5678   t       1
5678   y       2
5678   p       3

任何人都知道我該怎么做或者我應該使用什么 function？ 謝謝！

Answer 1

我相信您正在尋找groupby和cumcount() ，默認情況下+1從0開始：

df['Row number'] = df.groupby('Number').cumcount() + 1

print(df)

   Number Value  Row number
0    1234     a           1
1    1234     b           2
2    1234     x           3
3    5678     t           1
4    5678     y           2
5    5678     p           3