[python] PyYAML 사용 예시

PyYAML은 YAML(YAML Ain’t Markup Language) 형식의 데이터를 파이썬에서 해석하고 생성하기 위한 패키지입니다. YAML은 사람이 쉽게 읽고 쓸 수 있는 데이터 직렬화 형식으로 많은 프로그램들에서 사용됩니다. 이번 포스트에서는 PyYAML을 사용하여 YAML 데이터를 파이썬 객체로 읽고 쓰는 방법에 대해 알아보겠습니다.

설치하기

먼저 PyYAML을 설치해야 합니다. 다음 명령어를 사용하여 PyYAML을 설치할 수 있습니다:

pip install pyyaml

YAML 파일 읽기

import yaml

# YAML 파일 열기
with open('data.yaml') as file:
    # YAML 데이터를 파이썬 객체로 로드하기
    data = yaml.load(file, Loader=yaml.FullLoader)

    # 로드된 데이터 출력하기
    print(data)

위 예시에서는 yaml.load() 함수를 사용하여 YAML 파일을 파이썬 객체로 로드하였습니다. open() 함수를 사용하여 YAML 파일을 열고, with문을 사용하여 자동으로 파일을 닫습니다.

YAML 파일 쓰기

import yaml

# 파이썬 객체 생성하기
data = {'name': 'John Doe', 'age': 30, 'city': 'New York'}

# YAML 파일로 데이터 저장하기
with open('output.yaml', 'w') as file:
    # 파이썬 객체를 YAML 형식으로 직렬화하여 파일에 쓰기
    yaml.dump(data, file)

위 예시에서는 yaml.dump() 함수를 사용하여 파이썬 객체를 YAML 형식으로 직렬화하고, 파일에 씁니다. open() 함수를 사용하여 쓰기 모드로 파일을 열고, with문을 사용하여 자동으로 파일을 닫습니다.

결론

이처럼 PyYAML을 사용하면 YAML 형식의 데이터를 파이썬에서 쉽게 읽고 쓸 수 있습니다. YAML은 마크업보다 간단한 형식으로 데이터를 표현하는데 유용하며, PyYAML은 이를 처리하기 위한 효과적인 도구입니다.

참고 자료