[대기오염 예측모델 개발] 연도별 자치구별 대기오염물질 농도 히트맵 그리기

히트맵은 matplotlib, seaborn 두 가지 라이브러리를 이용해서 그릴 수 있는데 나는 seaborn을 사용했다.\

matplotlib보다 seaborn이 코드도 간결하고 더 예쁘게 나온다.

 

히트맵 그리는 순서

히트맵 그리는 순서는 다음과 같다.

1. 히트맵에 사용할 수 있도록 데이터 전처리(나는 groupby로 'year', 'location'별 대기오염물질 농도의 평균값을 구했다.)

df_train = df_train.groupby(['year', 'location'])[['no2', 'o3', 'co', 'so2', 'pm10', 'pm2.5']].mean().reset_index()

2. 히트맵을 그린다.

# 히트맵 그리기
plt.figure(figsize=(10, 20))
ax = sns.heatmap(df, annot = True, fmt = '.2f')
plt.title('Heatmap of no2', fontsize=16)
plt.show()

 

 

너무 간단한가..

 

히트맵 인자

annot	각 셀에 값을 표시할지의 여부. True로 설정하면 셀 안에 값을 넣는다.
camp	color map. 다양한 색상 맵 이름이 있는데, 대표적인 예는 'YlGnBu'이다. 색 순서를 반대로 하려면 'YlGnBu_r' 처럼 마지막에 '_r'을 붙여준다.
fmt	annot로 각 셀에 표시한 값을 어떤 형식으로 표시할지를 나타낸다. 나는 '.2f'로 해서 소수 둘째 자리까지 나타냈다.