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

Python Pandas – Seaborn 막대 그래프에서 중앙값을 중심 경향 추정치로 사용하는 방법

Seaborn의 막대 그래프(Bar Plot)는 점 추정값과 신뢰 구간을 사각형 막대 형태로 시각화하는 데 사용됩니다. 이를 위해 seaborn.barplot() 함수를 활용합니다. 데이터셋의 열을 x축과 y축 값으로 지정하면 손쉽게 가로 막대 그래프를 그릴 수 있으며, estimator 매개변수를 사용하면 중심 경향의 추정치를 기본값인 평균(mean) 대신 중앙값(median)으로 설정할 수 있습니다.

중앙값을 사용하면 극단적인 이상치(outlier)의 영향을 줄여 데이터의 실제 분포를 더 안정적으로 파악할 수 있다는 장점이 있습니다.

이번 예제에서는 다음과 같은 CSV 파일 형태의 데이터셋을 사용한다고 가정합니다 − Cricketers2.csv

1. 필요한 라이브러리 임포트

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

import seaborn as sb
import pandas as pd
import matplotlib.pyplot as plt
from numpy import median

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

CSV 파일의 데이터를 읽어 Pandas DataFrame에 저장합니다 −

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

3. 중앙값을 추정치로 사용한 가로 막대 그래프 그리기

Matches(경기 수)와 Academy(아카데미) 열을 사용해 가로 막대 그래프를 그리고, estimator 매개변수에 numpy의 median을 지정하여 중앙값을 중심 경향의 추정치로 설정합니다 −

sb.barplot(x = dataFrame["Academy"], y = dataFrame["Matches"], estimator = median)

전체 예제 코드

지금까지의 내용을 모두 합친 전체 코드는 다음과 같습니다 −

import seaborn as sb
import pandas as pd
import matplotlib.pyplot as plt
from numpy import median

# CSV 파일에서 데이터를 읽어 Pandas DataFrame에 저장
dataFrame = pd.read_csv("C:\\Users\\amit_\\Desktop\\Cricketers2.csv")

# Matches와 Academy 열로 가로 막대 그래프 그리기
# estimator 매개변수를 사용해 중앙값을 중심 경향의 추정치로 설정
sb.barplot(x = dataFrame["Academy"], y = dataFrame["Matches"], estimator = median)

# 그래프 출력
plt.show()

실행 결과

위 코드를 실행하면 아카데미별 경기 수의 중앙값이 막대 형태로 표시된 다음과 같은 그래프가 출력됩니다 −

Python Pandas – Seaborn 막대 그래프에서 중앙값을 중심 경향 추정치로 사용하는 방법