【pandas】重複のある要素に同じ連番の数字を振り直したい

Question

DataFrameで時系列データを整形するときに、重複のある要素に同じ連番の数字(ID)を振り直したいと考えているのですが、上手く対処するための方法が思いつかず、アドバイスをいただけないでしょうか。

下記に例を記載しました。実際には商品の後ろにたくさん列があるような形になっています。人と年齢は同一なので、この２要素が同じもにについては、同じ連番のIDを振りたいと思っています。どうぞ宜しくお願い致します

元のデータ

人	年齢	商品
a	30	A
b	20	B
a	30	C
d	40	D
d	40	E
f	61	F

最終的に作りたい形

人	年齢	商品	番号
a	30	A	1
b	20	B	2
a	30	C	1
d	40	D	3
d	40	E	3
f	61	F	4
。

Answer 1

pandas.core.groupby.GroupBy.ngroupが使えます。

import pandas as pd
from io import StringIO

s = """人,年齢,商品
a,30,A
b,20,B
a,30,C
d,40,D
d,40,E
f,61,F"""
df = pd.read_csv(StringIO(s))

df['番号'] = df.groupby(['人','年齢']).ngroup()+1
print(df)
#   人  年齢 商品  番号
#0  a  30  A   1
#1  b  20  B   2
#2  a  30  C   1
#3  d  40  D   3
#4  d  40  E   3
#5  f  61  F   4

Comments

[ROKUNO]

有難うございます、duplicatedなどで対応するような形で試して詰まってしまっており、groupbyの方法に辿り着けてませんでした。大変勉強になりました。