랭귀지/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)