파이썬으로 YAML 다루기: PyYAML 활용하기

파이썬에서 YAML(YAML Ain't Markup Language)은 데이터를 효율적으로 저장하고 전송하기 위한 형식 중 하나입니다. PyYAML은 파이썬에서 YAML을 다루는 데 사용되는 강력한 라이브러리입니다. 이 블로그에서는 PyYAML을 사용하여 YAML 파일을 읽고 쓰는 방법에 대해 알아보겠습니다.

1. YAML 파일 읽기

PyYAML을 사용하여 YAML 파일을 읽는 것은 간단합니다. 다음은 간단한 YAML 파일을 읽는 예제입니다.

import yaml

# YAML 파일 읽기
with open("example.yaml", "r") as file:
    data = yaml.load(file, Loader=yaml.FullLoader)

print(data)

위 예제는 "example.yaml" 파일을 읽고 그 내용을 파이썬 객체로 변환하여 출력합니다.

2. YAML 파일 쓰기

이번에는 PyYAML을 사용하여 YAML 파일을 생성하는 예제를 살펴보겠습니다.

import yaml

# 데이터 정의
data = {'name': 'John', 'age': 30, 'city': 'New York'}

# YAML 파일 쓰기
with open("output.yaml", "w") as file:
    yaml.dump(data, file)

print("YAML 파일이 생성되었습니다.")

위 예제는 파이썬 객체를 YAML 형식으로 변환하여 "output.yaml" 파일에 저장합니다.

3. YAML 내용 수정하기

기존의 YAML 파일을 읽어서 내용을 수정하고 다시 저장하는 방법을 살펴보겠습니다.

import yaml

# 기존 YAML 파일 읽기
with open("example.yaml", "r") as file:
    data = yaml.load(file, Loader=yaml.FullLoader)

# 데이터 수정
data['age'] = 35

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

print("YAML 파일이 수정되었습니다.")

위 예제는 기존의 YAML 파일을 읽어서 내용을 수정하고, 수정된 내용을 새로운 파일에 저장합니다.

파이썬으로 YAML을 다루는 방법에 대해 간단한 예제를 살펴보았습니다. PyYAML은 YAML 파일을 읽고 쓰는 작업을 간단하게 처리할 수 있도록 도와줍니다.