Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Matplotlib savefig() 함수에서 지원하는 파일 형식 목록을 확인하는 방법

Matplotlib의 savefig() 함수가 지원하는 파일 형식 목록을 확인하고 싶다면 get_supported_filetypes() 메서드를 사용하면 됩니다. 이 메서드는 현재 캔버스에서 저장 가능한 모든 파일 형식과 그에 대한 설명을 딕셔너리 형태로 반환해 줍니다.

실행 단계

  • plt.gcf()를 사용하여 현재 figure(그림) 객체를 가져옵니다.
  • 해당 figure를 포함하고 있는 캔버스(canvas)에 접근합니다.
  • 캔버스에서 get_supported_filetypes() 메서드를 호출합니다.
  • 반환된 파일 형식 항목을 반복문으로 순회하며 출력합니다.
  • 필요한 경우 plt.show() 메서드로 그래프를 화면에 표시할 수 있습니다.

예제 코드

from matplotlib import pyplot as plt

# 현재 figure의 캔버스에서 지원되는 파일 형식 가져오기
fs = plt.gcf().canvas.get_supported_filetypes()

# 파일 형식 항목 순회하며 출력
for key, val in fs.items():
    print(key, ":", val)

출력 결과

위 코드를 실행하면 다음과 같이 확장자(키)와 해당 형식의 설명(값)이 한 줄씩 출력됩니다.

eps : Encapsulated Postscript
jpg : Joint Photographic Experts Group
jpeg : Joint Photographic Experts Group
pdf : Portable Document Format
pgf : PGF code for LaTeX
png : Portable Network Graphics
ps : Postscript
raw : Raw RGBA bitmap
rgba : Raw RGBA bitmap
svg : Scalable Vector Graphics
svgz : Scalable Vector Graphics
tif : Tagged Image File Format
tiff : Tagged Image File Format

정리

이처럼 get_supported_filetypes() 메서드를 활용하면 Matplotlib 환경에서 그림을 저장할 때 사용할 수 있는 파일 형식을 손쉽게 파악할 수 있습니다. PNG, PDF, SVG, EPS 등 다양한 벡터 및 래스터 이미지 형식이 기본적으로 지원되므로, 용도에 맞는 형식을 선택하여 savefig('파일명.확장자') 형태로 그림을 저장하면 됩니다.