Python Pandas – Seaborn으로 수평 막대 그래프 그리기

Seaborn 막대 그래프(Bar Plot)란?

Seaborn의 막대 그래프는 각 범주에 대한 점 추정값(point estimate)과 신뢰 구간(confidence interval)을 사각형 막대 형태로 시각화하는 도구입니다. 특히 seaborn.barplot() 함수를 사용하면 손쉽게 수평 막대 그래프를 그릴 수 있습니다.

이번 글에서는 CSV 파일 형태의 데이터셋인 Cricketers2.csv(크리켓 선수 데이터)를 활용해, 아카데미별 경기 횟수를 비교하는 수평 막대 그래프를 만드는 과정을 단계별로 살펴보겠습니다.

1단계: 필요한 라이브러리 임포트

먼저 Seaborn, Pandas, Matplotlib 라이브러리를 임포트합니다.

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단계: 수평 막대 그래프 그리기

x축에 Matches(경기 횟수), y축에 Academy(아카데미)를 지정하여 barplot()을 호출합니다. y축에 범주형 변수가 위치하기 때문에 막대가 자동으로 수평 방향으로 그려집니다.

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

전체 예제 코드

지금까지의 과정을 하나로 정리한 전체 코드는 다음과 같습니다.

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")

# 수평 막대 그래프 그리기
sb.barplot(x = "Matches", y = "Academy", data= dataFrame)

# 그래프 화면에 표시
plt.show()

실행 결과

위 코드를 실행하면 다음과 같은 수평 막대 그래프가 출력됩니다. 각 막대의 길이는 해당 아카데미의 평균 경기 횟수를 나타내며, 검은색 오차 막대(error bar)는 신뢰 구간을 보여줍니다.