[python] Matplotlib에서 그래프 저장하기

Matplotlib는 파이썬에서 데이터 시각화를 위한 인기있는 라이브러리입니다. Matplotlib를 사용하여 생성한 그래프를 파일로 저장하는 방법에 대해 알아보겠습니다.

그래프 생성

우선, 간단한 라인 플롯을 생성해보겠습니다. 다음은 x와 y 값이 포함된 데이터를 사용하여 간단한 라인 플롯을 그리는 예제입니다.

import matplotlib.pyplot as plt

x = [1, 2, 3, 4, 5]
y = [2, 3, 5, 7, 11]

plt.plot(x, y)
plt.show()

그래프 저장

이제, 생성한 그래프를 이미지 파일로 저장해보겠습니다. Matplotlib를 사용하여 그래프를 저장하려면 savefig 함수를 사용합니다. 다음은 그래프를 PNG 파일로 저장하는 예제입니다.

plt.plot(x, y)
plt.savefig('line_plot.png')

위 코드를 실행하면 현재 작업 디렉토리에 line_plot.png 파일이 생성됩니다.

다양한 포맷으로 저장

savefig 함수를 사용하여 JPEG, PDF, SVG 등 다양한 포맷으로 그래프를 저장할 수 있습니다. 예를 들어, 다음과 같이 그래프를 PDF 파일로 저장할 수 있습니다.

plt.plot(x, y)
plt.savefig('line_plot.pdf')

결론

Matplotlib를 사용하여 생성한 그래프를 파일로 저장하는 방법을 알아보았습니다. savefig 함수를 활용하여 그래프를 원하는 포맷으로 저장할 수 있습니다.

이상으로 Matplotlib에서 그래프 저장하기에 대한 간단한 안내를 마치겠습니다.