랭귀지/pandas
pandas df와 df를 groupby sum한 a컬럼을 outer join하면 결과는?
유키공
2024. 8. 12. 19:59
import pandas as pd
# DataFrame 생성
data = {
'key': ['A', 'B', 'A', 'B', 'C'],
'value': [1, 2, 3, 4, 5]
}
df = pd.DataFrame(data)
# 'key'로 그룹화하고 'value' 컬럼의 합계 계산
grouped_sum = df.groupby('key')['value'].sum().reset_index()
grouped_sum.rename(columns={'value': 'total_value'}, inplace=True)
# 결과:
# key total_value
# 0 A 4
# 1 B 6
# 2 C 5
# 원래 DataFrame과 그룹화된 합계를 병합
df = df.merge(grouped_sum, on='key', how='left')
# 결과:
# key value total_value
# 0 A 1 4
# 1 B 2 6
# 2 A 3 4
# 3 B 4 6
# 4 C 5 5