iPython 노트북에서 Matplotlib로 생성한 그림(figure)을 이미지 파일로 저장하는 것은 데이터 분석 결과를 공유하거나 보고서에 활용할 때 매우 유용한 기능입니다. 몇 가지 간단한 단계만 거치면 손쉽게 그림을 파일로 저장할 수 있습니다.
그림을 파일로 저장하는 단계
- 새 그림(figure)을 생성하거나 기존 그림을 활성화합니다.
- add_axes() 메서드를 사용하여 그림에 축(axes)을 추가합니다.
- 주어진 리스트 데이터를 플롯(plot)합니다.
- savefig() 메서드를 사용하여 플롯을 파일로 저장합니다.
예제 코드
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
fig = plt.figure()
ax = fig.add_axes([1, 1, 1, 1])
plt.plot([1, 2])
plt.savefig('test.png', bbox_inches='tight')실행 결과
위 코드를 실행하면 다음과 같은 플롯이 test.png 파일로 저장됩니다.
여기서 bbox_inches='tight' 옵션은 그림 주변의 불필요한 여백을 자동으로 잘라내어 더 깔끔한 이미지를 얻을 수 있게 해줍니다. 또한 savefig() 메서드는 PNG뿐만 아니라 PDF, SVG, JPG 등 다양한 파일 형식을 지원하므로, 용도에 맞게 확장자를 변경하여 저장할 수 있습니다.