같은 인덱스 다른 값 리스트로 묶어주고 한 행으로 재배치하기

같은 인덱스 다른 값 리스트로 묶어주는 코드입니다.

 

import pandas as pd

# 예시용 데이터프레임 생성 
df = pd.DataFrame({"id" : [2011, 2011, 2012, 2012, 2013, 2013, 2013],
                  "value" : ['a', 'b', 'c', 'd', 'e', 'f', 'g']})
                  
# id를 인덱스로 설정 
df = df.set_index('id')

# id 별로 다른 값을 묶어서 리스트로 만든 후에 행 재배열
df.groupby('id').agg(lambda x: x.tolist())