[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 파일을 로딩하여 파이썬에서 사용할 수 있는 데이터 구조로 변환된 후 필요한 작업을 수행할 수 있습니다.