[python] PyYAML을 사용하여 YAML 파일의 데이터를 찾아 바꾸기

30 Nov 2023

YAML은 인간이 읽기 쉽고 작성하기 쉬운 데이터 직렬화 형식입니다. YAML 파일을 다룰 때 데이터를 읽고 수정해야 할 때가 있습니다. 이때 PyYAML 라이브러리를 사용하면 간단하게 데이터를 찾아 바꿀 수 있습니다.

PyYAML 설치하기

PyYAML을 사용하기 위해서는 먼저 해당 라이브러리를 설치해야 합니다. 아래의 명령어를 사용하여 PyYAML을 설치하세요.

pip install pyyaml

먼저 YAML 파일을 읽어와야 합니다. yaml.load() 함수를 사용하여 YAML 파일의 데이터를 파이썬 객체로 로드할 수 있습니다.

import yaml

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

위의 코드에서 'example.yaml'은 읽을 YAML 파일의 경로입니다. yaml.load() 함수를 사용하여 파일을 읽어와 data 변수에 저장했습니다.

파일을 읽어와서 데이터를 찾는 방법은 다양합니다. 데이터를 찾는 방법은 YAML 파일의 구조에 따라 다르지만, 일반적으로 키 및 값을 이용하여 탐색할 수 있습니다.

key = 'name'

if key in data:
    value = data[key]
    print(value)
else:
    print(f'{key} not found')

위의 코드에서는 key 변수에 원하는 키 값을 지정한 후 해당 키가 data 변수에 있는지 확인합니다. 키가 있다면 해당 키에 해당하는 값을 얻고 출력합니다. 키가 없다면 해당 키가 존재하지 않음을 출력합니다.

PyYAML을 사용하여 YAML 파일의 데이터를 변경하는 방법은 간단합니다. 데이터를 변경한 후 yaml.dump() 함수를 사용하여 변경된 데이터를 다시 YAML 형식으로 변환한 다음 파일에 저장하면 됩니다.

key = 'name'

if key in data:
    data[key] = 'New Name'

with open('example.yaml', 'w') as file:
    yaml.dump(data, file)

위의 코드에서는 key 값을 기준으로 데이터를 변경하고, 변경된 데이터를 다시 example.yaml 파일에 저장했습니다.

PyYAML을 사용하면 YAML 파일의 데이터를 쉽게 찾고 변경할 수 있습니다. 위에서 제시한 코드를 참고하여 YAML 파일을 읽고 데이터를 찾고 변경해 보세요.