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

Python Pandas – Seaborn에서 order 매개변수로 순서를 지정해 바이올린 플롯 그리기

Seaborn 바이올린 플롯(Violin Plot)이란?

Seaborn의 바이올린 플롯은 박스플롯(boxplot)커널 밀도 추정(Kernel Density Estimate)을 하나로 결합한 시각화 기법입니다. 이를 그릴 때는 seaborn.violinplot() 함수를 사용하며, order 매개변수를 활용하면 범주형 데이터의 표시 순서를 명시적으로 지정할 수 있습니다.

이번 예제에서는 크리켓 선수 정보가 담긴 CSV 파일(Cricketers.csv)을 데이터셋으로 사용합니다.

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

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

2. CSV 파일을 Pandas DataFrame으로 불러오기

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

3. 명시적 순서를 지정하여 바이올린 플롯 그리기

'Role'(역할)과 'Age'(나이)를 기준으로 바이올린 플롯을 그려보겠습니다. 이때 order 매개변수에 원하는 순서를 직접 전달하면 "Role" 항목의 표시 순서를 자유롭게 제어할 수 있습니다.

sb.violinplot(x = 'Role', y = "Age", order=["Batsman", "Bowler"], 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\\Cricketers.csv")

# Role(역할)과 Age(나이)로 바이올린 플롯 그리기
# order 매개변수를 사용해 "Role" 기준의 순서를 명시적으로 지정
sb.violinplot(x = 'Role', y = "Age", order=["Batsman", "Bowler"], data = dataFrame)

# 그래프 출력
plt.show()

실행 결과

위 코드를 실행하면 다음과 같은 바이올린 플롯이 화면에 나타납니다.

Python Pandas – Seaborn에서 order 매개변수로 순서를 지정해 바이올린 플롯 그리기