[Python] YAML 파일 사용하기

python에서 yaml 파일을 사용하기 위해서 yaml 모듈을 설치해주면된다.

 

import yaml

로 사용하는 모듈은 설치시에 다음과 같이 해주면 된다.

 

pip install PyYAML

그냥 pip install yaml로 하면 모듈을 찾을 수 없게 되니 참고 바랍니다.

 

yaml 파일을 쓰는 방법은 다음과 같습니다.

 

import yaml

with open('test.yaml') as f:
    data = yaml.load(f)
    
a = data['a_col']
b = data['b_col']

yaml 모듈은 yaml 파일을 python 딕셔너리 형태로 만들어줍니다. 참고하세요.