[python+matplotlib] matplotlib.pyplot으로 만든 plot 그림 파일로 저장하기, savefig()

관련 글

[python] matlab과 비슷한 방식으로 plot 그리기, matplotlib.pyplot

 

 

matplotlib 패키지의 pyplot을 이용하면 손쉽게 plot을 그릴 수 있습니다. 오늘은 그렇게 생성한 plot을 그림 파일로 저장하는 법에 대해 살펴보겠습니다. 보고서나 논문을 쓸 때 그래프를 가져다가 쓰려면 그림 파일로 저장해 줄 필요가 있겠죠? 

 

먼저 sin 함수에 대한 그래프를 간단히 그리겠습니다. 코드와 생성된 그래프는 각각 다음과 같습니다. 

 

import numpy as np
from matplotlib import pyplot as plt
import math
 
x = np.arange(0, 4*math.pi, 0.1)
y = np.sin(x)
 
plt.plot(x, y)

 

 

 y = sin(x)를 0부터 4파이까지 두 주기 그려봤습니다. 

 

이제 이 그래프를 그림 파일로 저장해보도록 하겠습니다. matplotlib.pyplot.savefig 함수를 사용하면 됩니다. 다음과 같이 저는 한 줄의 코드를 덧붙여 줌으로 그래프를 .png 파일로 저장했습니다. 참고로 .bmp 파일은 지원하지 않습니다. 

 

import numpy as np
from matplotlib import pyplot as plt
import math
 
x = np.arange(0, 4*math.pi, 0.1)
y = np.sin(x)
 
plt.plot(x, y)
 
plt.savefig('./sin.png')

 

위 코드를 실행하면 .py 파일이 존재하는 디렉토리 내에 sin.png 그림 파일이 생성됩니다. 

 

저장 완료!