아미(아름다운미소)

pandas df와 df를 groupby sum한 a컬럼을 outer join하면 결과는? 본문

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