블로그

[파이썬] matplotlib 서버에서 실행시 그래프 이미지로 저장

파이썬에서 matplotlib 를 사용해서 그래프를 그렸는데 서버의 터미널에서 실행할 경우

확인이 어려운 경우가 있습니다.

이런 경우는 그래프를 이미지 파일로 저장 후 다운로드 받아서 확인이 가능합니다.

아래는 텐서플로우에서 간단히 선 그래프를 그리는 샘플입니다.

import tensorflow as tf
import matplotlib
matplotlib.use('Agg') # pyplot 을 import 하기 전에 먼저 해 줘야 합니다.
import matplotlib.pyplot as plt

fig = plt.figure()

x = range(20)
y = tf.random.normal([20],0,1)
plt.plot(x,y)

fig.savefig('1.png')

이렇게 해서 저장된 이미지 파일은 아래와 같습니다.

enter image description here