파이썬에서 그룹별 합계와 평균 구하기

파이썬에서 그룹별로 합계와 평균을 구하는 법은 pandas라는 모듈을 사용하면 빠르게 구할 수 있습니다. 이번 포스팅에서는 코드 몇 줄로 그룹별 합계와 평균을 구하는 법에 대해서 알아보겠습니다.

 

 

목차

1. 1. 서론
2. 2. pandas 라이브러리 소개
3. 3. groupby 함수 사용법
   1. 3.1 그룹별 합계 구하기
   2. 3.2 그룹별 평균 구하기
4. 4. agg 함수 사용법
5. 5. 결론

1. 서론

파이썬은 데이터 분석에 가장 널리 사용되는 프로그래밍 언어 중 하나입니다. 특히, pandas라는 라이브러리는 데이터프레임이라는 강력한 데이터 구조를 제공하여, 복잡한 데이터 연산을 간결하게 수행할 수 있게 돕습니다. 이번 섹션에서는 파이썬의 pandas 라이브러리를 사용하여 그룹별 합계와 평균을 계산하는 방법을 소개하겠습니다.

 

2. pandas 라이브러리 소개

pandas는 데이터 분석을 위한 파이썬 라이브러리입니다. pandas의 핵심 구조는 데이터프레임으로, 2차원의 테이블 데이터를 효율적으로 처리할 수 있습니다. 다양한 데이터 타입을 다루며, 행과 열을 기준으로 데이터를 선택하거나 조작하는 등의 기능을 제공합니다.

 

3. groupby 함수 사용법

pandas의 groupby 함수는 특정 열을 기준으로 데이터를 그룹화하고, 각 그룹에 대해 다양한 연산을 수행할 수 있습니다.

 

3.1 그룹별 합계 구하기

import pandas as pd
df = pd.DataFrame({'group': ['A', 'A', 'B', 'B', 'C', 'C'], 'value': [1, 2, 3, 4, 5, 6]})
sum_df = df.groupby('group')['value'].sum()
print(sum_df)

 

3.2 그룹별 평균 구하기

mean_df = df.groupby('group')['value'].mean()
print(mean_df)

 

4. agg 함수 사용법

agg 함수는 여러 개의 함수를 동시에 적용하거나 그룹별로 다른 함수를 적용할 때 사용합니다.

agg_df = df.groupby('group')['value'].agg(['sum', 'mean'])
print(agg_df)

5. 결론

파이썬의 pandas 라이브러리를 통해 쉽게 그룹별 합계와 평균을 구할 수 있습니다. groupby와 agg 함수를 적절히 활용하여 다양한 그룹 연산을 수행해 보세요.