[python] Seaborn을 사용한 에러 바 그래프 그리기

데이터 시각화는 데이터의 패턴과 관계를 보여주는 강력한 도구입니다. Seaborn 라이브러리는 Matplotlib을 기반으로 한 Python 데이터 시각화 라이브러리로, 아름다운 그래프를 쉽게 그릴 수 있습니다. 이번에는 Seaborn을 사용하여 에러 바 그래프를 그려보겠습니다.

에러 바 그래프란 데이터 포인트의 평균과 표준 오차를 시각적으로 나타내는 그래프입니다. 일반적으로 실험 데이터를 분석할 때, 그룹 간의 차이를 검정하기 위해 사용됩니다.

필요한 라이브러리 설치하기

먼저 Seaborn 라이브러리를 설치해야 합니다. 아래 명령을 사용하여 설치할 수 있습니다.

pip install seaborn

예제 코드

Seaborn을 사용하여 에러 바 그래프를 그리는 간단한 예제 코드를 먼저 살펴보겠습니다. 아래 코드는 3개의 그룹(A, B, C)에 대한 실험 데이터를 가지고 에러 바 그래프를 그리는 예제입니다.

import seaborn as sns
import matplotlib.pyplot as plt

# 실험 데이터
group_A = [10, 12, 9, 11, 12]
group_B = [8, 9, 11, 7, 10]
group_C = [6, 7, 8, 9, 10]

# 그래프 설정
sns.set(style="whitegrid")

# 그래프 그리기
sns.barplot(x=["A", "B", "C"], y=[np.mean(group_A), np.mean(group_B), np.mean(group_C)], 
            ci="sd", capsize=0.2, palette="Blues_d")

# 그래프에 에러 바 추가하기
plt.errorbar(x=["A", "B", "C"], y=[np.mean(group_A), np.mean(group_B), np.mean(group_C)], 
             yerr=[np.std(group_A), np.std(group_B), np.std(group_C)], fmt="none", color="black", capsize=0.2)

# 그래프 타이틀 설정
plt.title("Error Bar Graph")

# 그래프 보여주기
plt.show()

코드 설명

  1. Seaborn과 Matplotlib을 import 합니다.
  2. 실험 데이터를 변수로 정의합니다.
  3. 그래프 스타일을 설정합니다.
  4. sns.barplot()을 사용하여 바 그래프를 그립니다. 이 때 ci 파라미터를 "sd"로 설정하여 표준 편차를 에러 바로 표시하도록 합니다.
  5. plt.errorbar()을 사용하여 에러 바를 추가합니다. yerr 파라미터에 각 그룹의 표준 편차를 입력하고 fmt 파라미터를 "none"으로 설정하여 선을 그리지 않도록 합니다.
  6. 그래프의 타이틀을 설정합니다.
  7. 그래프를 보여줍니다.

실행 결과

위의 예제 코드를 실행하면 아래와 같은 에러 바 그래프가 그려집니다.

에러 바 그래프

마무리

Seaborn을 사용하면 손쉽게 에러 바 그래프를 그릴 수 있습니다. 이를 통해 실험 데이터를 비교하고 그룹 간의 차이를 시각화할 수 있습니다. Seaborn의 다양한 기능과 스타일을 활용하여 데이터 시각화를 더욱 효과적으로 할 수 있습니다.

더 많은 Seaborn 기능과 그래프 예제는 Seaborn 공식 문서를 참고하시기 바랍니다.