Pandas 그룹 내에서 이전 행과의 차이 구하기

1 개요[ | ]

Pandas 컬럼 이전 행과의 차이 구하기

import pandas as pd
import numpy as np
import io

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

df['diff'] = df['num'] - df.groupby('user_id')['num'].shift()
df

# 정수형 (NaN은 0으로 처리)
temp = df['num'] - df.groupby('user_id')['num'].shift()
df['diff2'] = np.where(temp.isna(), 0, temp).astype('int')
df

2 같이 보기[ | ]