Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Matplotlib.pyplot으로 만든 그래프를 이미지 파일로 저장하는 방법

Matplotlib.pyplot으로 이미지 저장하기

Python의 대표적인 데이터 시각화 라이브러리인 Matplotlib에서는 savefig() 함수를 사용해 화면에 그린 그래프를 이미지 파일로 손쉽게 저장할 수 있습니다. 아래 단계를 순서대로 따라 해 보세요.

이미지 저장 절차

  • 그림 크기 설정 — figure 크기를 지정하고 서브플롯 사이와 주변의 여백(padding)을 조정합니다.
  • 데이터 생성 — NumPy를 이용해 x축과 y축에 사용할 데이터 포인트를 만듭니다.
  • 그래프 그리기 — plot() 메서드로 데이터를 시각화합니다.
  • 파일 저장 — savefig() 메서드를 호출해 그림을 원하는 파일명으로 저장합니다.

예제 코드

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

x = np.linspace(-np.pi, np.pi, 100)

plt.plot(x, np.sin(x) * x, c='red')

plt.savefig("myimage.png")

실행 결과

위 코드를 실행하면 그려진 그래프가 프로젝트 폴더에 "myimage.png"라는 이름의 PNG 파일로 자동 저장됩니다.

Matplotlib.pyplot으로 만든 그래프를 이미지 파일로 저장하는 방법

참고: 알아두면 좋은 savefig() 옵션

  • dpi — 이미지 해상도를 지정합니다. 예: plt.savefig("myimage.png", dpi=300)
  • format — 'png', 'pdf', 'svg', 'jpg' 등 다양한 파일 형식을 지원합니다.
  • bbox_inches='tight' — 불필요한 여백을 잘라내어 꽉 찬 이미지로 저장합니다.
  • transparent=True — 배경을 투명하게 처리한 이미지를 저장할 수 있습니다.