파이썬에서 그래프를 이미지 파일로 저장하기

문제 상황: plt.show() 대신 이미지로 저장하기

파이썬의 Matplotlib 라이브러리를 사용하여 그래프를 그릴 때, 일반적으로 plt.show() 함수를 이용해 화면에 그래프를 출력합니다. 그러나 이 방법으로는 그래프를 파일로 저장할 수 없습니다. 이 문제를 어떻게 해결할 수 있을까요?

해결 방법: plt.savefig()

Matplotlib 라이브러리에서 제공하는 plt.savefig() 함수를 사용하면 이 문제를 쉽게 해결할 수 있습니다. 이 함수는 그래프를 다양한 형식의 이미지 파일로 저장할 수 있게 해줍니다.

사용 예시

import matplotlib.pyplot as plt

# 그래프를 그립니다.
plt.plot([1, 2, 3], [4, 5, 1])

# 그래프를 파일로 저장합니다.
plt.savefig('plot.png')

위의 코드를 실행하면, 현재 작업 디렉토리에 ‘plot.png’라는 이름의 이미지 파일이 생성됩니다.

주의할 점

1. 파일 형식: plt.savefig() 함수는 다양한 파일 형식을 지원합니다. 예를 들어, .png, .jpg, .svg 등을 사용할 수 있습니다.
2. 경로 설정: 저장 경로를 지정하지 않으면 현재 작업 디렉토리에 파일이 저장됩니다. 경로를 지정하려면 문자열로 경로를 입력해주면 됩니다.
3. 함수 순서: plt.savefig()는 plt.show() 이전에 호출되어야 합니다. plt.show() 이후에 호출하면 빈 그래프가 저장됩니다.

기타 옵션: 화질과 크기 조절

plt.savefig() 함수는 추가적인 매개변수를 통해 이미지의 화질이나 크기를 조절할 수 있습니다. 예를 들어, dpi 매개변수를 사용하면 이미지의 해상도를 설정할 수 있습니다.

화질 조절 예시

plt.savefig('plot_high_quality.png', dpi=300)

이 옵션을 사용하면 해상도가 300dpi인 고화질 이미지로 저장됩니다.

정리

Matplotlib 라이브러리의 plt.savefig() 함수를 이용하면, 그래프를 이미지 파일로 쉽게 저장할 수 있습니다. 다양한 옵션을 통해 원하는 형식과 화질로 조절할 수 있어 매우 유용합니다.