Seaborn은 파이썬의 데이터 시각화 라이브러리 중 하나로, Matplotlib을 기반으로 한다. Seaborn은 Matplotlib보다 좀 더 간편하고 세련된 스타일을 제공하여, 통계적인 그래프를 그리기에 편리하다. 이번 포스트에서는 Seaborn을 사용하여 트리 플롯을 그리는 방법에 대해 알아보겠다.

필요한 패키지 설치하기

먼저, Seaborn 패키지를 설치해야한다. 아래의 명령어를 사용하여 설치할 수 있다.

pip install seaborn

샘플 데이터셋 불러오기

Seaborn은 내장된 샘플 데이터셋을 제공하여 테스트 용도로 활용할 수 있다. 이번 예제에서는 타이타닉 데이터셋을 사용할 예정이다. 아래의 코드를 사용하여 데이터셋을 불러오자.

import seaborn as sns

titanic_data = sns.load_dataset('titanic')

트리 플롯 그리기

트리 플롯은 변수들 사이의 관계를 시각적으로 표현하는 그래프이다. Seaborn을 사용하여 트리 플롯을 그리기 위해서는 sns.clustermap() 함수를 사용한다. 아래의 코드는 타이타닉 데이터셋의 생존 여부에 따른 성별, 요금, 선실 등급을 시각화한 트리 플롯의 예제이다.

import seaborn as sns
import matplotlib.pyplot as plt

sns.clustermap(titanic_data[['survived', 'sex', 'fare', 'class']])
plt.show()

위의 코드를 실행하면, 생존 여부에 따른 성별, 요금, 선실 등급에 대한 트리 플롯이 그려진다.

결론

Seaborn을 사용하여 트리 플롯을 그리는 방법에 대해 알아보았다. Seaborn은 Matplotlib보다 좀 더 간편하고 세련된 스타일을 제공하여, 통계적인 그래프를 그리기에 매우 편리하다. 다양한 시각화 기법을 사용하여 데이터의 분포 및 관계를 탐색하는데 활용할 수 있다.

참고 자료

• Seaborn 공식 문서