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

Seaborn으로 하나의 그래프에 두 개의 바이올린 플롯 시리즈 그리는 방법

Seaborn에서 hue 매개변수를 활용하면 하나의 그래프에 두 개의 바이올린 플롯(violin plot) 시리즈를 손쉽게 나란히 그릴 수 있습니다. 바이올린 플롯은 박스 플롯과 커널 밀도 추정(KDE)을 결합한 시각화 기법으로, 데이터의 분포 형태를 직관적으로 보여준다는 장점이 있습니다.

구현 단계

  • 그림 크기를 설정하고 서브플롯 사이 및 주변의 여백(padding)을 조정합니다.

  • 온라인 저장소에서 예제 데이터셋을 불러옵니다(인터넷 연결이 필요합니다).

  • violinplot() 메서드를 사용하여 바이올린 플롯을 생성합니다. 이때 hue 인자에 범주형 변수를 지정하면 각 x축 값마다 두 개의 바이올린이 색상으로 구분되어 나란히 그려집니다.

  • plt.show() 메서드를 호출하여 완성된 그래프를 화면에 표시합니다.

예제 코드

# Seaborn과 Matplotlib 임포트
import seaborn as sns
from matplotlib import pyplot as plt

# 그림 크기 설정
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True

# 예제 데이터셋 불러오기
tips = sns.load_dataset("tips")

# Seaborn으로 바이올린 플롯 생성
sns.violinplot(x="day", y="total_bill", hue="time", data=tips)

# 그래프 표시
plt.show()

코드 설명

위 코드에서는 Seaborn에 내장된 tips 데이터셋을 사용합니다. x축에는 요일(day), y축에는 총 지출액(total_bill)이 배치되며, hue="time" 옵션 덕분에 점심(Lunch)과 저녁(Dinner)이라는 두 개의 시리즈가 동일한 그래프 안에서 서로 다른 색상으로 구분되어 나란히 표시됩니다. 이처럼 hue 매개변수만 잘 활용하면 별도의 그래프를 만들지 않고도 두 시리즈를 한눈에 비교할 수 있습니다.

실행 결과

위 코드를 실행하면 다음과 같은 출력 결과를 얻을 수 있습니다 −

Seaborn으로 하나의 그래프에 두 개의 바이올린 플롯 시리즈 그리는 방법