[파이썬 캐글뽀개기] Matplotlib - 히트 맵(Heat Map)

[파이썬 캐글뽀개기] Matplotlib - 히트 맵(Heat Map)

• 다양한 강도, 색상으로 데이터 범위를 시각화
• ex. 상관행렬의 히트맵 표시
• matplotlib의 Imshow 함수 이용

#heat map
import matplotlib.pyplot as plt
import numpy as np
import seaborn as sns

flights = sns.load_dataset('flights')
flights = flights.pivot("month","year","passengers")

fig, ax = plt.subplots(figsize=(12,6))
im = ax.imshow(flights, cmap='YlGnBu')
ax.set_xticklabels(flights.columns, rotation=20)
ax.set_yticklabels(flights.index, rotation=10)
fig.colorbar(im)

fig.show()

 

f