[Python] 그룹별 산점도 점 색깔과 모양 다르게 하기 (Scatter Plot by Groups) (2/4)

이번 포스팅은 두 개의 연속형 변수에 대한 관계를 파악하는데 유용하게 사용할 수 있는 산점도(Scatter Plot) 의 두번째 포스팅으로서 그룹별로 산점도의 점의 색깔과 모양을 다르게 하는 방법을 소개합니다. 

 

(1) 산점도 (Scatter Plot)

(2) 그룹별 산점도 (Scatter Plot by Groups)

(3) 산점도의 marker 크기 및 색깔, 모양 설정 (Setting Marker's size, color, shape)

(4) 산점도 행렬 (Scatter Plot Matrix)

 

 

산점도를 그리는데 사용할 데이터는 iris 로서, 'petal length'와 'petal width'의 연속형 변수에 대해서 'species' 그룹별로 점의 색깔과 모양을 다르게 설정해보겠습니다. 

 

참고로 species 에는 setosa 50개, versicolor 50개, virginica 50개씩의 관측치가 들어있습니다. 

 

 

# importing libraries import numpy as np import pandas as pd   import matplotlib.pyplot as plt import seaborn as sns plt.rcParams['figure.figsize'] = [10, 8] # setting figure size

 

# loading 'iris' dataset from seaborn iris = sns.load_dataset('iris') iris.shape (150, 5)   iris.head()   sepal_length sepal_width petal_length petal_width species 0 5.1 3.5 1.4 0.2 setosa 1 4.9 3.0 1.4 0.2 setosa 2 4.7 3.2 1.3 0.2 setosa 3 4.6 3.1 1.5 0.2 setosa 4 5.0 3.6 1.4 0.2 setosa   iris.groupby('species').size() species setosa 50 versicolor 50 virginica 50 dtype: int64

 

 

 

(1) matplotlib 으로 그룹별 산점도 그리기 (scatter plot by groups via matplotlib)

 

# Scatter plot with a different color by groups groups = iris.groupby('species')   fig, ax = plt.subplots() for name, group in groups:     ax.plot(group.petal_length,              group.petal_width,              marker='o',              linestyle='',             label=name) ax.legend(fontsize=12, loc='upper left') # legend position plt.title('Scatter Plot of iris by matplotlib', fontsize=20) plt.xlabel('Petal Length', fontsize=14) plt.ylabel('Petal Width', fontsize=14) plt.show()

 

 

 

(2) seaborn 으로 그룹별 산점도 그리기 (scatter plot by groups via seaborn)

 

코드가 깔끔하고 가독성이 좋으며, 산점도 그래프도 보기에 참 좋습니다. 

 

# Scatter plot by Groups sns.scatterplot(x='petal_length',                  y='petal_width',                  hue='species', # different colors by group                 style='species', # different shapes by group                 s=100, # marker size                 data=iris) plt.show()

 

 

 

혹시 AttributeError: module 'seaborn' has no attribute 'scatterplot' 에러가 나면 seaborn version upgrade를 해주시기 바랍니다. 

 

# error: AttributeError: module 'seaborn' has no attribute 'scatterplot'  # solution: upgrade seaborn pip install --upgrade seaborn

 

 

 

(3) pandas로 그룹별 산점도 그리기 (scatter plot by groups via pandas)

# adding 'color' column iris['color'] = np.where(iris.species == 'setosa', 'red',                           np.where(iris.species =='versicolor',                           'green',                           'blue'))   # scatter plot iris.plot(kind='scatter',           x='petal_length',            y='petal_width',            s=50, # marker size           c=iris['color']) # marker color by group   plt.title('Scatter Plot of iris by pandas', fontsize=20) plt.xlabel('Petal Length', fontsize=14) plt.ylabel('Petal Width', fontsize=14) plt.show()

 

 

많은 도움이 되었기를 바랍니다. 

이번 포스팅이 도움이 되었다면 아래의 '공감~

'를 꾹 눌러주세요. ^^

 

다음번 포스팅에서는 산점도의 marker 크기 및 색깔, 모양 설정 (Setting Marker's size, color, shape) 에 대해서 소개하겠습니다.