[python] PyYAML로 YAML 파일을 읽어와서 일부 데이터 변경하기

YAML은 인간이 쉽게 읽을 수 있는 데이터 직렬화 형식입니다. YAML 파일을 읽어와서 일부 데이터를 변경하고 싶은 경우, Python의 PyYAML 라이브러리를 사용할 수 있습니다. 이 블로그 포스트에서는 PyYAML을 사용하여 YAML 파일을 읽어와서 일부 데이터를 변경하는 방법에 대해 알아보겠습니다.

필요한 패키지 설치

먼저, PyYAML 패키지를 설치해야 합니다. 아래의 명령어를 사용하여 설치할 수 있습니다.

pip install pyyaml

YAML 파일 읽기

PyYAML을 사용하여 YAML 파일을 읽기 위해서는 yaml.safe_load() 함수를 사용해야 합니다. 아래의 예제 코드는 test.yaml 파일을 읽어와서 Python 변수에 저장하는 예제입니다.

import yaml

with open('test.yaml', 'r') as file:
    data = yaml.safe_load(file)

데이터 변경하기

이제 YAML 파일에서 일부 데이터를 변경하고 싶을 때, Python 변수를 수정하고 다시 YAML 파일로 저장할 수 있습니다. 아래의 예제 코드는 YAML 파일에서 ‘name’ 키의 값을 변경하는 예제입니다.

data['name'] = 'New Name'

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

전체 코드

import yaml

# YAML 파일 읽기
with open('test.yaml', 'r') as file:
    data = yaml.safe_load(file)

# 데이터 변경하기
data['name'] = 'New Name'

# YAML 파일 저장
with open('test.yaml', 'w') as file:
    yaml.dump(data, file)

위의 코드를 실행하면 ‘test.yaml’ 파일에서 ‘name’ 키의 값이 ‘New Name’으로 변경되어 저장됩니다.

PyYAML을 사용하여 YAML 파일을 읽어와서 일부 데이터를 변경하는 방법에 대해 알아보았습니다. 이를 응용하여 원하는 데이터를 YAML 파일에서 수정하는 작업을 수행할 수 있습니다. 추가적인 사용법은 PyYAML 공식 문서를 참고하시기 바랍니다.

참고 문서: PyYAML 공식 문서