[python] PyYAML으로 YAML 파일을 읽어와서 그래프로 시각화하기 (matplotlib, seaborn 등 사용)

30 Nov 2023

이번 포스트에서는 PyYAML 라이브러리를 사용하여 YAML 파일을 읽어와서 그래프로 시각화해 보겠습니다. 그래프를 그리기 위해 matplotlib과 seaborn 라이브러리도 함께 사용하겠습니다.

PyYAML 설치하기

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

pip install pyyaml

다음은 data.yaml 파일을 읽어오는 예제입니다.

import yaml

with open('data.yaml', 'r') as file:
    data = yaml.safe_load(file)

위 코드는 data.yaml 파일을 읽어와서 data 변수에 저장합니다.

이제 데이터를 시각화해 보겠습니다. 여기서는 matplotlib과 seaborn을 함께 사용하여 그래프를 그립니다.

import matplotlib.pyplot as plt
import seaborn as sns

# 데이터 프레임 생성
df = pd.DataFrame(data)

# 그래프 그리기
sns.barplot(x='x', y='y', data=df)
plt.show()

위 코드는 데이터를 이용하여 바 그래프를 그립니다. x와 y는 데이터의 칼럼 이름을 나타내며, df는 데이터 프레임을 의미합니다.

이렇게 PyYAML을 사용하여 YAML 파일을 읽어와서 그래프로 시각화하는 방법을 알아보았습니다. PyYAML 라이브러리를 통해 YAML 파일을 간편하게 처리할 수 있으며, matplotlib과 seaborn 라이브러리를 사용하여 그래프를 그릴 수 있습니다.