[python] PyYAML을 이용하여 YAML 파일의 데이터를 분석하여 그래프로 표현하기

30 Nov 2023

python

YAML은 인간이 쉽게 읽고 쓸 수 있는 데이터 직렬화 양식입니다. 이러한 YAML 파일을 파이썬에서 쉽게 분석하여 그래프로 표현하는 방법을 알아보겠습니다.

PyYAML 설치하기

PyYAML은 파이썬에서 YAML 파일을 다루기 위해 사용하는 라이브러리입니다. 다음 명령어를 사용하여 PyYAML을 설치할 수 있습니다.

pip install pyyaml

YAML 파일 분석하기

다음은 분석할 YAML 파일의 예시입니다.

- name: Alice
  age: 25
  profession: Programmer
- name: Bob
  age: 30
  profession: Designer

import yaml

# YAML 파일 열기
with open('data.yaml', 'r') as file:
    # YAML 데이터 로드
    data = yaml.load(file, Loader=yaml.FullLoader)

# 데이터 출력
for item in data:
    print(f"Name: {item['name']}")
    print(f"Age: {item['age']}")
    print(f"Profession: {item['profession']}")

위의 코드에서는 yaml.load() 함수를 사용하여 YAML 파일의 데이터를 로드합니다. 이때 Loader=yaml.FullLoader를 지정하여 보안 문제를 방지할 수 있습니다.

그래프로 표현하기

파이썬에서 그래프로 표현하기 위해서는 matplotlib 라이브러리를 사용할 수 있습니다. 다음은 데이터를 바탕으로 그래프를 그리는 예시입니다.

import yaml
import matplotlib.pyplot as plt

# YAML 파일 열기
with open('data.yaml', 'r') as file:
    # YAML 데이터 로드
    data = yaml.load(file, Loader=yaml.FullLoader)

# 그래프 데이터
names = [item['name'] for item in data]
ages = [item['age'] for item in data]

# 그래프 그리기
plt.bar(names, ages)
plt.xlabel('Name')
plt.ylabel('Age')
plt.title('Age of People')
plt.show()

위의 코드에서는 names와 ages 변수에 각각 이름과 나이 데이터를 저장한 뒤, plt.bar() 함수를 사용하여 그래프를 그립니다. 그래프의 레이블과 제목은 plt.xlabel(), plt.ylabel(), plt.title() 함수로 설정할 수 있습니다.

이제 위의 코드를 실행하면, YAML 파일의 데이터를 그래프로 표현할 수 있습니다.

결론

PyYAML 라이브러리를 사용하여 YAML 파일의 데이터를 파이썬에서 분석하고, 해당 데이터를 그래프로 표현하는 방법을 알아보았습니다. 이를 통해 YAML 파일의 데이터를 시각적으로 표현함으로써 더 쉽게 이해할 수 있습니다.

참고 자료

PyYAML 공식 문서: https://pyyaml.org/
matplotlib 공식 문서: https://matplotlib.org/