[python] PyYAML로 YAML 파일을 읽어와서 JSON으로 변환하기

이번 블로그 포스트에서는 Python에서 PyYAML 라이브러리를 사용하여 YAML 파일을 읽어와서 JSON 형식으로 변환하는 방법에 대해 알아보겠습니다.

필요한 라이브러리 설치

먼저 PyYAML 라이브러리를 설치해야 합니다. 아래 명령을 사용하여 설치할 수 있습니다.

pip install pyyaml

YAML 파일 읽기

다음으로, YAML 파일을 읽어와야 합니다. YAML 파일에는 텍스트 형식으로 데이터가 저장되어 있으며, PyYAML 라이브러리를 사용하여 파이썬에서 이를 읽어올 수 있습니다. 아래 코드를 사용하여 YAML 파일을 읽어올 수 있습니다.

import yaml

def read_yaml(filename):
    with open(filename, 'r') as file:
        data = yaml.safe_load(file)
        return data

yaml_data = read_yaml('example.yaml')

위 코드에서 read_yaml 함수는 yaml.safe_load 함수를 사용하여 YAML 파일을 읽어옵니다. 읽어온 데이터는 변수 yaml_data에 저장됩니다. example.yaml은 읽어올 YAML 파일의 경로와 파일명으로 변경해야 합니다.

JSON으로 변환하기

YAML 파일을 읽어온 후, 이를 JSON 형식으로 변환해야 합니다. 다음 코드를 사용하여 YAML 데이터를 JSON 형식으로 변환할 수 있습니다.

import json

def convert_to_json(yaml_data):
    json_data = json.dumps(yaml_data, indent=4)
    return json_data

json_data = convert_to_json(yaml_data)

위 코드에서 convert_to_json 함수는 json.dumps 함수를 사용하여 YAML 데이터를 JSON 형식으로 변환합니다. 변환된 데이터는 변수 json_data에 저장됩니다.

JSON 파일로 저장하기

마지막으로, 변환된 JSON 데이터를 파일로 저장할 수 있습니다. 아래 코드를 사용하여 JSON 데이터를 파일로 저장할 수 있습니다.

def save_json(json_data, filename):
    with open(filename, 'w') as file:
        file.write(json_data)

save_json(json_data, 'example.json')

위 코드에서 save_json 함수는 변환된 JSON 데이터를 지정된 파일에 저장합니다. example.json은 저장할 JSON 파일의 경로와 파일명으로 변경해야 합니다.

정리

이번 포스트에서는 PyYAML 라이브러리를 사용하여 YAML 파일을 읽어와서 JSON 형식으로 변환하는 방법에 대해 알아보았습니다. 이를 통해 YAML과 JSON 간의 데이터 변환이 간단하게 이루어질 수 있습니다.

참고 자료