[python] PyYAML로 YAML 파일 로딩하기

많은 데이터를 다루는 프로그램에서는 설정 파일이나 데이터 파일을 사용하는 경우가 많습니다. 이러한 파일 형식 중 하나인 YAML 파일은 사람이 쉽게 읽고 쓰기 위해 설계된 것으로, Python에서는 PyYAML 라이브러리를 사용하여 YAML 파일을 로딩하고 파싱할 수 있습니다.

PyYAML 설치하기

먼저, PyYAML 라이브러리를 설치해야 합니다. pip 패키지 매니저를 사용하여 다음과 같이 설치할 수 있습니다.

pip install pyyaml

PyYAML을 사용하여 YAML 파일 로딩하기

PyYAML 라이브러리를 사용하여 YAML 파일을 로딩하는 방법은 매우 간단합니다. 아래는 예시 YAML 파일을 로딩하는 코드입니다.

import yaml

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

print(data)

위 코드에서 open 함수를 사용하여 YAML 파일을 읽고, yaml.load 함수를 사용하여 파일을 파싱합니다. yaml.load 함수의 첫 번째 매개변수로는 열려 있는 파일 객체, 두 번째 매개변수로는 Loader를 지정합니다. yaml.FullLoader는 모든 YAML 스펙을 지원하는 로더를 사용하는 것을 의미합니다.

YAML 파일 구조

로드된 데이터는 YAML 파일의 구조와 마찬가지로 딕셔너리와 리스트의 중첩 구조를 가지게 됩니다. 예를 들어, 아래와 같은 YAML 파일이 있다고 가정해봅시다.

# example.yaml

fruits:
  - apple
  - orange
  - banana

colors:
  - red
  - green
  - blue

위 YAML 파일을 로딩한 데이터는 다음과 같은 구조를 가지게 됩니다.

{
    'fruits': ['apple', 'orange', 'banana'],
    'colors': ['red', 'green', 'blue']
}

로드된 데이터를 이용하여 원하는 작업을 수행할 수 있습니다.

결론

PyYAML 라이브러리를 사용하여 YAML 파일을 로딩하는 것은 간단하고 효과적입니다. YAML 파일을 로딩하여 파이썬에서 사용할 수 있는 데이터 구조로 변환된 후 필요한 작업을 수행할 수 있습니다.