[python] PyYAML을 사용하여 YAML 파일의 데이터를 다른 포맷으로 변환하기 (Excel, PDF 등)

YAML은 인간이 쉽게 읽고 쓸 수 있는 데이터 직렬화 포맷입니다. 그러나 때로는 YAML 파일의 데이터를 다른 포맷으로 변환해야 할 수도 있습니다. 이번 블로그 포스트에서는 PyYAML 라이브러리를 사용하여 YAML 파일의 데이터를 Excel, PDF 등 다른 포맷으로 변환하는 방법을 알아보겠습니다.

1. PyYAML 설치

PyYAML을 사용하기 위해 먼저 해당 라이브러리를 설치해야 합니다. 다음 명령어를 사용하여 설치할 수 있습니다:

pip install pyyaml

2. YAML 파일 읽기

먼저 YAML 파일을 읽어와야 합니다. 다음 예제 코드는 “data.yaml” 파일을 읽어옵니다:

import yaml

with open('data.yaml', 'r') as file:
    data = yaml.safe_load(file)

3. 데이터 변환

이제 YAML 파일의 데이터를 원하는 포맷으로 변환할 수 있습니다.

3.1 Excel로 변환

YAML 파일의 데이터를 Excel 파일로 변환하는 방법은 다음과 같습니다:

import pandas as pd

df = pd.DataFrame(data)
df.to_excel('data.xlsx', index=False)

위 코드를 실행하면 “data.xlsx”라는 파일이 생성되며, 해당 파일에 YAML 파일의 데이터가 엑셀 시트로 저장됩니다.

3.2 PDF로 변환

YAML 파일의 데이터를 PDF로 변환하기 위해서는 다음과 같은 절차가 필요합니다:

  1. YAML 데이터를 HTML 형식으로 변환합니다.
  2. HTML을 PDF로 변환합니다.
from yaml import dump
from pdfkit import from_file

# YAML 데이터를 HTML로 변환
html = f'<pre>{dump(data)}</pre>'

# HTML을 PDF로 변환
from_file(html, 'data.pdf')

위 코드를 실행하면 “data.pdf”라는 파일이 생성되며, 해당 파일에 YAML 파일의 데이터가 PDF로 저장됩니다.

마무리

이제 PyYAML을 사용하여 YAML 파일의 데이터를 Excel, PDF 등 다른 포맷으로 변환하는 방법을 알아보았습니다. 이를 통해 YAML 파일의 데이터를 다른 양식에서 쉽게 사용할 수 있게 되었습니다.

이 글이 유용하게 되었길 바랍니다!

참고 자료