Seaborn을 사용하여 고급 플롯 만들기

20 Oct 2023

Seaborn은 파이썬의 시각화 라이브러리로서, Matplotlib를 기반으로한 고급 플롯을 쉽게 만들 수 있게 해줍니다. 이번 튜토리얼에서는 Seaborn을 사용하여 고급 플롯을 만드는 방법을 알아보겠습니다.

필요한 패키지 설치

Seaborn을 사용하기 위해서는 우선 Seaborn 패키지를 설치해야 합니다. 아래의 명령을 사용하여 설치할 수 있습니다.

pip install seaborn

Seaborn은 내장 데이터셋을 제공해줍니다. 이번 예제에서는 ‘tips’라는 데이터셋을 사용하겠습니다. 데이터를 로드하는 코드는 다음과 같습니다.

import seaborn as sns

tips_data = sns.load_dataset('tips')

Seaborn을 사용하여 히스토그램을 그리는 방법은 간단합니다. 다음의 코드를 사용하여 ‘total_bill’ 컬럼의 값들에 대한 히스토그램을 그릴 수 있습니다.

sns.histplot(data=tips_data, x='total_bill')

두 변수 사이의 관계를 살펴보기 위해 Seaborn을 사용하여 산점도를 그릴 수 있습니다. 아래의 코드는 ‘total_bill’과 ‘tip’ 두 변수 사이의 산점도를 그리는 예시입니다.

sns.scatterplot(data=tips_data, x='total_bill', y='tip')

데이터의 변수들 간의 상관관계를 시각화하기 위해 Seaborn의 히트맵을 사용할 수 있습니다. 다음의 코드는 각 변수들 간의 상관관계를 히트맵으로 그리는 예시입니다.

sns.heatmap(data=tips_data.corr(), annot=True)

Seaborn으로 그린 플롯을 이미지 파일로 저장하려면 savefig 함수를 사용할 수 있습니다. 다음의 코드는 히스토그램을 이미지 파일로 저장하는 예시입니다.

sns.histplot(data=tips_data, x='total_bill')
plt.savefig('histogram.png')

#python #datavisualization