[python] PyYAML을 사용하여 YAML 파일의 데이터를 특정 조건에 따라 수정하기
목차
YAML은 읽기와 쓰기가 쉬운 형식으로 데이터를 저장하는 포맷입니다. PyYAML은 YAML 파일의 데이터를 파싱하고, Python 데이터 구조로 변환할 수 있는 라이브러리입니다. 이번 글에서는 PyYAML을 사용하여 YAML 파일의 데이터를 특정 조건에 맞게 수정하는 방법을 알아보겠습니다.
YAML 파일 불러오기
먼저, PyYAML 패키지를 설치해야 합니다. 다음 명령을 사용하여 설치할 수 있습니다:
pip install pyyaml
YAML 파일을 불러오기 위해 다음과 같은 코드를 작성합니다:
import yaml
# YAML 파일 불러오기
with open('data.yaml', 'r') as file:
data = yaml.load(file, Loader=yaml.FullLoader)
위 코드에서 data.yaml는 수정하고자 하는 YAML 파일의 경로를 나타냅니다.
YAML 파일 수정하기
YAML 파일의 데이터를 수정하려면, 불러온 데이터를 조작하면 됩니다. 예를 들어, 특정 조건에 따라 데이터를 수정해보겠습니다:
# 데이터 수정하기
for item in data:
if item['price'] > 100:
item['discounted_price'] = item['price'] * 0.9
else:
item['discounted_price'] = item['price']
위 코드는 데이터의 ‘price’ 키 값이 100보다 크다면 ‘discounted_price’ 키를 추가하고, 할인 가격을 계산하여 저장합니다.
YAML 파일 저장하기
수정된 데이터를 다시 YAML 파일로 저장하려면 다음과 같은 코드를 사용합니다:
# YAML 파일 저장하기
with open('modified_data.yaml', 'w') as file:
yaml.dump(data, file)
위 코드에서 modified_data.yaml는 수정된 데이터를 저장할 YAML 파일의 경로를 나타냅니다.
이제 모든 작업이 완료되었습니다. PyYAML을 사용하여 YAML 파일의 데이터를 특정 조건에 맞게 수정하는 방법을 알아보았습니다.