아미(아름다운미소)

pandas transform 본문

랭귀지/pandas

pandas transform

유키공 2024. 8. 28. 08:28
import pandas as pd

# 샘플 데이터프레임 생성
data = {
    '팀': ['A', 'A', 'B', 'B', 'C', 'C'],
    '점수': [10, 20, 30, 40, 50, 60]
}

df = pd.DataFrame(data)

# 각 팀의 점수 평균을 계산하고 원래 데이터프레임에 추가
df['점수_평균'] = df.groupby('팀')['점수'].transform('mean')

print(df)
Comments