python pandas 重複要素の連番付与について

Question

以下のようなデータフレームにて、重複した数値を持つIDがあります。
このIDを元に、列を追加して、重複したIDごとに連番をつけたいのですが
イメージがつかないため、教えていただけると幸いです。

ID	時
25433	13
25433	13
25433	13
111931	7
111931	14

実現したいこと
→同じIDに対して連番の値を返す。

ID	時	count
25433	13	1
25433	13	2
25433	13	3
111931	7	1
111931	14	2

Answer 1

GroupBy.cumcount() をお使いください

import pandas as pd

df = pd.DataFrame({
  'ID':[25433,25433,25433,11931,11931],
  '時':[13,13,13,7,14]})

df['count'] = df.groupby('ID').cumcount()+1
print(df)
#      ID   時  count
#0  25433  13      1
#1  25433  13      2
#2  25433  13      3
#3  11931   7      1
#4  11931  14      2

Comments

[ABSS]

ありがとうございます！想定した結果になりました！