Matplotlib에서 Pandas 데이터프레임 그래프를 PDF 파일로 저장하는 방법

Pandas 데이터프레임을 기반으로 만든 그래프를 PDF 파일로 내보내려면 다음 단계를 따르면 됩니다.

주요 단계

• 그림(figure) 크기를 설정하고, 서브플롯 주변 및 사이의 여백(padding)을 조정합니다.
• 세 개의 열(col1, col2, col3)을 가진 Pandas 데이터프레임을 생성합니다.
• plot() 메서드를 사용하여 데이터프레임을 시각화합니다.
• 그래프를 화면에 표시하려면 show() 메서드를 사용합니다.

예제 코드

import matplotlib.pyplot as plt
import pandas as pd

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

df = pd.DataFrame([[2, 1, 4], [5, 2, 1], [4, 0, 1]], columns=['col1', 'col2', 'col3'])
df.plot()

plt.savefig('pd_df.pdf')

실행 결과

위 코드를 실행하면 생성된 그래프가 "pd_df.pdf"라는 이름의 PDF 파일로 저장됩니다. 핵심은 savefig() 함수에 파일명을 지정할 때 확장자를 .pdf로 작성하는 것입니다. Matplotlib는 확장자를 인식하여 자동으로 해당 형식의 파일로 변환해 저장해 줍니다.

이 방법은 PNG, SVG 등 다른 이미지 형식으로도 동일하게 적용할 수 있으며, 벡터 기반인 PDF로 저장하면 확대해도 그래프의 품질이 저하되지 않는다는 장점이 있습니다.