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

파이썬에서 YAML 파일을 읽어와서 데이터를 추출하고 정렬하는 방법을 알아보겠습니다. 이를 위해 PyYAML 라이브러리를 사용할 것입니다.

PyYAML 설치하기

먼저, PyYAML을 설치해야 합니다. 다음 명령을 사용하여 PyYAML을 설치할 수 있습니다.

pip install pyyaml

YAML 파일 읽기

아래의 코드를 사용하여 YAML 파일을 읽어옵니다.

import yaml

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

위의 코드에서 data.yaml은 읽어올 YAML 파일의 경로입니다. 읽어온 데이터는 data 변수에 저장됩니다.

데이터 추출하기

추출할 데이터에 따라 코드를 조정할 수 있습니다. 아래는 예시입니다.

# YAML 파일에서 특정 데이터 추출하기
name = data['name']
age = data['age']
skills = data['skills']

# 추출한 데이터 출력하기
print(f"이름: {name}")
print(f"나이: {age}")
print(f"기술: {skills}")

데이터 정렬하기

데이터를 정렬하기 위해 sorted() 함수를 사용할 수 있습니다. 아래는 예시 코드입니다.

# skills 리스트를 알파벳순으로 정렬하기
sorted_skills = sorted(skills)

# 정렬된 데이터 출력하기
print(f"정렬된 기술: {sorted_skills}")

이제 위 코드를 사용하여 YAML 파일의 데이터를 추출하고 정렬할 수 있습니다.

참고 자료