Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Python Pandas - Seaborn으로 막대 그래프를 그리고 표준 편차 표시하기

Seaborn의 막대 그래프(Bar Plot)는 점 추정값과 신뢰 구간을 사각형 막대 형태로 시각화하는 데 사용됩니다. 이를 위해 seaborn.barplot() 함수를 활용하며, 관측값의 표준 편차를 표시하려면 신뢰 구간(confidence interval)을 설정하는 ci 매개변수에 "sd" 값을 지정하면 됩니다.

이번 예제에서는 CSV 파일 형태의 다음 데이터셋(Cricketers2.csv)을 사용하겠습니다.

1. 필요한 라이브러리 가져오기

먼저 필요한 라이브러리들을 임포트합니다.

import seaborn as sb
import pandas as pd
import matplotlib.pyplot as plt

2. CSV 파일에서 데이터 불러오기

Pandas의 read_csv() 함수를 사용해 CSV 파일의 데이터를 DataFrame으로 불러옵니다.

dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\Cricketers2.csv")

3. 막대 그래프 그리기 및 표준 편차 표시

Academy(아카데미)와 Matches(경기 수) 열을 기준으로 막대 그래프를 그리고, ci 매개변수에 "sd" 값을 지정하여 관측값의 표준 편차를 함께 표시합니다.

sb.barplot(x = "Academy", y = "Matches", data = dataFrame, ci = "sd")

전체 예제 코드

다음은 위 과정을 모두 포함한 전체 코드입니다.

import seaborn as sb
import pandas as pd
import matplotlib.pyplot as plt

# CSV 파일에서 데이터를 읽어 Pandas DataFrame으로 불러오기
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\Cricketers2.csv")

# Seaborn 테마 설정
sb.set_theme(style="darkgrid")

# Academy와 Matches로 막대 그래프 그리기
# ci 매개변수에 "sd"를 지정하여 관측값의 표준 편차 표시
sb.barplot(x = "Academy", y = "Matches", data = dataFrame, ci = "sd")

# 그래프 출력
plt.show()

실행 결과

위 코드를 실행하면 각 아카데미별 경기 수 평균과 함께 오차 막대(error bar)로 표준 편차가 표시된 막대 그래프가 출력됩니다.

Python Pandas - Seaborn으로 막대 그래프를 그리고 표준 편차 표시하기

참고 사항

Seaborn 최신 버전(v0.12 이상)에서는 ci 매개변수가 deprecated되었으며, 대신 errorbar=("sd", 1)과 같이 errorbar 매개변수를 사용하는 것이 권장됩니다.