Python 그림을 표시하는 대신 이미지 파일로 플롯을 저장합니다., Save plot to image file instead of displaying it

질문

이것은 GUI에서 도표를 표시합니다:

import matplotlib.pyplot as plt
plt.plot([1, 2, 3], [1, 4, 9])
plt.show()

하지만 대신에 도표를 파일로 저장하는 방법은 어떻게 하나요 (예 : foo.png)?

답변

matplotlib.pyplot.savefig을(를) 사용할 때, 파일 형식은 확장자로 지정할 수 있습니다:

from matplotlib import pyplot as plt

plt.savefig('foo.png')
plt.savefig('foo.pdf')

이렇게 하면 각각 래스터화 된 또는 벡터화 된 출력물이 생성됩니다. 또한 이미지 주변에 불필요한 여백이 있을 수 있으므로 다음과 같이 제거할 수 있습니다:

plt.savefig('foo.png', bbox_inches='tight')

그림을 보여줄 경우, plt.show()는 plt.savefig() 다음에 따라와야 합니다. 그렇지 않으면 파일 이미지가 공백이 됩니다.