YAML 파일은 데이터를 구조적으로 저장하는 형식으로 널리 사용됩니다. PyYAML은 파이썬에서 YAML 파일을 처리하는 기능을 제공하는 라이브러리입니다. 이 블로그 포스트에서는 PyYAML을 사용하여 YAML 파일에서 특정 데이터를 제거하고, 그 자리에 새로운 데이터를 추가하는 방법을 알아보겠습니다.

YAML 파일에서 데이터 제거하기

먼저, PyYAML을 설치해야 합니다. 다음 명령어를 사용하여 PyYAML을 설치합니다.

pip install pyyaml

설치가 완료되면, YAML 파일을 로드하고 데이터를 제거하는 방법을 알아보겠습니다.

import yaml

# YAML 파일 로드
with open('data.yaml', 'r') as file:
    data = yaml.safe_load(file)

# 데이터 제거
del data['key1']

# 수정된 데이터를 YAML 파일에 저장
with open('data.yaml', 'w') as file:
    yaml.dump(data, file)

위 코드에서 key1이라는 키에 해당하는 데이터를 제거하였습니다. yaml.safe_load() 함수를 사용하여 YAML 파일을 로드하고, 변경된 데이터를 yaml.dump() 함수로 다시 YAML 파일에 저장합니다.

YAML 파일에 데이터 추가하기

이제, YAML 파일에 새로운 데이터를 추가하는 방법을 알아보겠습니다.

import yaml

# YAML 파일 로드
with open('data.yaml', 'r') as file:
    data = yaml.safe_load(file)

# 새로운 데이터 추가
data['key2'] = 'value2'

# 수정된 데이터를 YAML 파일에 저장
with open('data.yaml', 'w') as file:
    yaml.dump(data, file)

위 코드에서는 key2라는 키에 value2라는 값을 추가하였습니다. 마찬가지로 yaml.safe_load() 함수와 yaml.dump() 함수를 사용하여 데이터를 로드하고 저장합니다.

결론

PyYAML을 이용하여 YAML 파일에서 특정 데이터를 제거하고 새로운 데이터를 추가하는 방법을 알아보았습니다. 이를 활용하여 YAML 파일을 유연하게 처리할 수 있습니다.

참고 문서: PyYAML 공식 문서