그림 11-12 역 이름, 비율 수치 표시, 색상 변경을 반영한 파이 차트

마지막으로 시각화한 자료를 이미지 파일로 저장합니다. 지금은 그래프가 나타나는 창마다 하나씩 저장 버튼을 눌러 이미지 파일로 저장해야 해서 번거롭습니다. 역마다 자동으로 각 그래프 자료가 이미지 파일로 저장되면 좋겠네요.

그래프를 이미지로 저장하려면 savefig() 함수를 사용하면 됩니다. 이때 이미지 파일의 이름과 형식을 지정할 수 있습니다. row[3]에는 역 이름, row[1]에는 노선 이름이 저장되어 있습니다. 따라서 역 이름과 노선 이름을 넣어 파일 제목으로 두고, png 형식의 이미지 파일로 저장하고 싶다면 row[3] + ' ' + row[1] + '.png' 형태로 이미지 파일 이름을 지정할 수 있습니다. 그러면 파일 이름이 ‘강남 2호선.png’처럼 만들어질 것입니다.

다음 코드만 추가하면 됩니다.

plt.axis('equal')
plt.savefig(row[3]+' '+row[1]+'.png')  # 이미지 파일로 저장
plt.show()

다음은 전체 코드입니다.