Pandas 그룹 내에서 인접한 것끼리 그룹 지정하기

1 개요

Pandas 그룹 내에서 인접한 것끼리 그룹 지정하기

2 예시 1

• user_id별로 num이 연속된 값이면 같은 그룹, 그렇지 않으면 다른 그룹

import pandas as pd
import io

tsv="""
user_id	num
2	3
2	4
1	4
2	5
1	5
2	6
1	7
2	8
1	8
1	9
"""
df = pd.read_csv(io.StringIO(tsv),sep='\t')
df

df['group'] = df.groupby('user_id')['num'].diff() > 1
df['group'] = df.user_id*100 + df.groupby('user_id')['group'].cumsum() + 1
df

5 같이 보기

• Pandas groupby()
• Pandas cumsum()
• Pandas 인접한 것끼리 그룹 지정하기
• Pandas 그룹 내에서 이전 행과의 차이 구하기

6 참고

• https://stackoverflow.com/questions/41620920/groupby-conditional-sum-of-adjacent-rows-pandas