Matplotlib를 활용하면 몇 가지 간단한 단계만으로 모자이크 플롯(mosaic plot)을 손쉽게 그릴 수 있습니다. 모자이크 플롯은 범주형 데이터의 비율과 상대적 크기를 직사각형 영역으로 표현하여 한눈에 비교할 수 있게 해주는 유용한 시각화 기법입니다.
모자이크 플롯 작성 단계
그림(figure) 크기를 설정하고, 서브플롯 사이 및 주변의 여백(padding)을 조정합니다.
statsmodels 패키지를 설치합니다(
pip install statsmodels). 모자이크 플롯을 생성하려면 이 패키지가 반드시 필요합니다. statsmodels는 기술 통계(descriptive statistics)뿐만 아니라 통계 모델의 추정과 추론까지 지원하는 통계 계산용 Python 패키지로, scipy를 보완하는 역할을 합니다.모자이크 플롯에 사용할 데이터를 딕셔너리(dictionary) 형태로 준비합니다.
준비된 교차표(contingency table) 데이터를 기반으로 모자이크 플롯을 생성합니다.
show() 메서드를 호출하여 완성된 그림을 화면에 표시합니다.
예제 코드
import matplotlib.pyplot as plt
from statsmodels.graphics.mosaicplot import mosaic
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
# 모자이크 플롯용 딕셔너리
data = {'John': 7, 'Joe': 10, 'James': 5, 'Kate': 1}
# 모자이크 플롯 생성
mosaic(data, title='Basic mosaic plot')
# 그림 표시
plt.show()
실행 결과
위 코드를 실행하면 다음과 같은 모자이크 플롯이 출력됩니다.
