[python] PyYAML을 사용하여 YAML 파일의 데이터를 압축하고 새로운 파일로 저장하기

이번 블로그 포스트에서는 Python의 PyYAML 라이브러리를 사용하여 YAML 파일의 데이터를 압축하고 새로운 파일로 저장하는 방법을 알아보겠습니다.

필요한 라이브러리 설치하기

먼저, PyYAML 라이브러리를 설치해야 합니다. 아래 명령어를 사용하여 설치할 수 있습니다.

pip install pyyaml

YAML 파일 읽기

압축하고자 하는 YAML 파일을 읽어와야 합니다. yaml 모듈을 가져와서 파일을 열고, yaml.load() 함수를 사용하여 데이터를 읽습니다.

import yaml

with open('input.yaml', 'r') as file:
    data = yaml.load(file, Loader=yaml.FullLoader)

input.yaml은 압축하고자 하는 원본 YAML 파일의 경로입니다.

YAML 데이터 압축하기

PyYAML을 사용하여 YAML 데이터를 압축하려면 yaml.dump() 함수를 사용합니다. 이 함수는 첫 번째 인자로 압축할 데이터를 받고, 두 번째 인자로 압축된 데이터를 저장할 파일을 받습니다.

compressed_data = yaml.dump(data, allow_unicode=True)

allow_unicode=True는 유니코드 문자를 포함한 데이터를 처리하기 위한 옵션입니다.

압축된 데이터 새로운 파일로 저장하기

압축된 데이터를 새로운 파일로 저장할 수 있습니다. yaml.dump() 함수의 두 번째 인자로 파일을 지정하여 데이터를 저장합니다.

with open('output.yaml', 'w') as file:
    yaml.dump(data, file, allow_unicode=True)

output.yaml은 압축된 데이터를 저장할 파일의 경로입니다.

전체 코드

import yaml

# YAML 파일 읽기
with open('input.yaml', 'r') as file:
    data = yaml.load(file, Loader=yaml.FullLoader)

# YAML 데이터 압축하기
compressed_data = yaml.dump(data, allow_unicode=True)

# 압축된 데이터 새로운 파일로 저장하기
with open('output.yaml', 'w') as file:
    yaml.dump(data, file, allow_unicode=True)

위의 코드를 실행하면 input.yaml 파일의 데이터가 압축되어 output.yaml 파일로 저장됩니다.

결론

이제 PyYAML 라이브러리를 사용하여 YAML 파일의 데이터를 압축하고 새로운 파일로 저장하는 방법을 알게 되었습니다. 이를 통해 YAML 데이터를 효율적으로 관리하고 저장할 수 있습니다.