[Python] matplotlib.pyplot.savefig 파일 저장 시 여백 제거 방법

👨‍💻 1.  개요

본 포스팅에서는 파이썬 matplotlib.pyplot 패키지를 활용한 시각화 자료 저장 시 자료 내 불필요한 여백을 제거하는 방법에 대해 다룹니다.

🤖 2.  문제점

아래 (그림 1)과 같이, matplotlib 옵션 설정 없이 시각화 자료를 저장하게 되면 그래프 주변 사방으로 불필요한 여백이 많은 것을 확인할 수 있습니다. 이는 보고서나 대시보드 등에 활용 시 다른 차트들과의 심미성 있는 조화를 방해하는 요소가 되기도 합니다.

그림 1. 시각화 시 불필요한 흰 바탕 여백

 

[예시 코드 - savefig 메소드 옵션 미설정]

# 패키지 설치 필요 : pip install matplotlib
import matplotlib.pyplot as plt

# 그래프 생성
plt.plot([1, 2, 3, 4])
plt.ylabel('some numbers')

# 그래프 저장
plt.savefig('./my_figure.png')

✅ 3.  해결 방법

해결 방법은 심플하게 savefig 메소드의 bbox_inches 옵션 값으로 'tight' 만 설정해 주시면 됩니다. 

아래 (그림 2 우측)과 같이, 불필요한 여백이 제거된 것을 확인하실 수 있습니다.

그림 2. bbox_inches 옵션 tight 설정여부별 저장 결과 비교

 

[솔루션 코드]

# 그래프 저장
plt.savefig('./my_figure_tight.png', bbox_inches='tight')

---

피드백 환영합니다.

그럼 오늘도 즐거운 하루 보내세요 :)

고맙습니다.