[PYTHON] GridSpec 을 이용한 여러 그래프를 같이 그리기

0. 목표

 - GridSpec 을 이용한 여러 그래프를 같이 그리기

1. 실습

 1) library 호출

import seaborn as sns
import matplotlib.pyplot as plt
from matplotlib import gridspec

 2) 데이터 생성

flights_long = sns.load_dataset("flights")

 3) heatmap 용 데이터

flights = flights_long.pivot("month", "year", "passengers")

 4) barplot 용 데이터

year_df = flights_long.groupby(by = 'year').agg({'passengers' : 'sum'})
month_df = flights_long.groupby(by = 'month').agg({'passengers' : 'sum'})

year_df.reset_index(inplace = True)
month_df.reset_index(inplace = True)

 5) 그림 그리기

# 도화지
fig = plt.figure(figsize = (15, 15))
gs = gridspec.GridSpec(nrows = 2,  
                       ncols = 2,  
                       height_ratios = [5, 10], 
                       width_ratios = [10, 5]
                      )
fig.subplots_adjust(wspace = 0, hspace = 0)
ax0 = plt.subplot(gs[0])
ax2 = plt.subplot(gs[2])
ax3 = plt.subplot(gs[3])

# barplot1
sns.barplot(x = "year", y = "passengers", data = year_df, ax = ax0)
ax0.set_ylabel('passengers', fontsize = 25)
ax0.set_xticks([])
ax0.tick_params(axis = 'y', labelsize = 15)

# heatplot
sns.heatmap(flights, annot=True, fmt="d", linewidths=.5, ax=ax2, cbar = False, cmap="Blues", annot_kws={"size": 15})
ax2.set_xlabel('year', fontsize = 25)
ax2.set_ylabel('month', fontsize = 25)
ax2.tick_params(axis = 'x', labelsize = 15)
ax2.tick_params(axis = 'y', labelsize = 15)

# barplot2
sns.barplot(x = "passengers", y = "month", data = month_df, ax = ax3)
ax3.set_yticks([])
ax3.set_ylabel('')
ax3.set_xlabel('passengers', fontsize = 25)
ax3.tick_params(axis = 'x', labelsize = 15)

  - 결과

2. 참고

 - subplot의 사이즈를 각각 다르게 조절합시다!

 - Annotated heatmaps

 - seaborn.heatmap

 - seaborn.barplot

 - [seaborn] heatmap 경계선 굵기 설정, 글자 크기 조절