데이터 시각화 첫걸음 떼기

이제 matplotlib 패키지를 사용해 폴더 크기 데이터를 시각화해 봅시다.

비주얼 스튜디오 코드에서 새로운 파일을 생성한 뒤 파일명을 'step3_2.py'로 변경합니다. 다음과 같이 코드를 입력하고, 대화형 창에서 실행하세요. 실행 결과를 'output' 폴더에 'step_3_2.png' 파일을 생성됩니다.

from pathlib import Path
import matplotlib.pyplot as plt # 2
from step_2_1 import OUT_DIR # 이전에 작성한 모듈운 불러옵니다.
from step_3_1 import load_plot_data

plot_data = load_plot_data()
fig, ax = plt.subplots() # 7
ax.barh(plot_data["stem"], plot_data["size"]) # 8
fig.savefig(OUT_DIR / f"{Path(__file__).stem}.png") # 9

 

[실행결과]

02 matplotlib 패키지의 pyplot 모듈을 불러옵니다.

07 plt 모듈의 함수 subplots()를 호출하여 객체 Figure와 Axes를 생성하고 각각 변수 fig, ax에 저장합니다.

08 함수 barh()를 호출하여 바 차트를 그립니다. 첫 번째 입력값으로 세로축에 사용할 폴더 이름을, 두 번째 입력값으로 가로축에 사용할 폴더 크기를 전달합니다.

09 함수 함수 savefig()를 호출하여 fig에 저장된 차트를 png 파일로 저장합니다.