[python] PyYAML을 사용하여 YAML 파일의 데이터 정렬하기

YAML은 데이터를 인간이 읽고 쓰기 쉽도록 표현하는 형식입니다. PyYAML은 Python에서 YAML 파일을 처리하기 위한 라이브러리입니다. YAML 파일의 데이터를 정렬하는 것은 가독성과 유지 보수를 위해 중요합니다. 이번 포스트에서는 PyYAML을 사용하여 YAML 파일의 데이터를 정렬하는 방법에 대해 알아보겠습니다.

PyYAML 설치하기

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

pip install pyyaml

YAML 파일 읽기

YAML 파일을 읽기 위해서는 yaml 모듈을 임포트해야 합니다. 아래의 예제는 example.yaml 파일을 읽고 해당 데이터를 출력하는 코드입니다.

import yaml

with open('example.yaml', 'r') as file:
    data = yaml.load(file, Loader=yaml.Loader)

print(data)

데이터 정렬하기

PyYAML은 sort_keys 옵션을 지원하여 데이터를 정렬할 수 있습니다. sort_keys 옵션을 True로 설정하면, 딕셔너리 키를 알파벳 순으로 정렬합니다. 아래의 예제는 데이터를 읽은 후 sort_keys 옵션을 사용하여 데이터를 정렬한 뒤 출력하는 코드입니다.

import yaml

with open('example.yaml', 'r') as file:
    data = yaml.load(file, Loader=yaml.Loader)
    sorted_data = yaml.dump(data, sort_keys=True)

print(sorted_data)

YAML 파일 쓰기

정렬된 데이터를 YAML 파일로 쓰기 위해서는 yaml.dump() 함수를 사용합니다. 아래의 예제는 정렬된 데이터를 output.yaml 파일로 쓰는 코드입니다.

import yaml

with open('example.yaml', 'r') as file:
    data = yaml.load(file, Loader=yaml.Loader)
    sorted_data = yaml.dump(data, sort_keys=True)

with open('output.yaml', 'w') as file:
    file.write(sorted_data)

결론

PyYAML을 사용하여 YAML 파일의 데이터를 정렬하는 방법에 대해 알아보았습니다. 데이터를 정렬함으로써 YAML 파일의 가독성을 높이고 유지 보수를 용이하게 할 수 있습니다. PyYAML의 다양한 옵션을 사용하여 YAML 파일을 처리하는 방법을 익혀보세요.

참고 자료