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

Python 분석과 프로그래밍/Python 그래프_시각화 2019.01.13 18:56

이번 포스팅은 두 개의 연속형 변수에 대한 관계를 파악하는데 유용하게 사용할 수 있는 산점도(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) 에 대해서 소개하겠습니다.

저작자표시비영리변경금지

'Python 분석과 프로그래밍 > Python 그래프_시각화' 카테고리의 다른 글

[Python] 산점도 행렬 (Scatterplot Matrix) (4/4) (0)	2019.01.15
[Python] 4개 변수로 점의 크기와 색깔을 다르게 산점도 그리기 (Scatter plot with 4 variables, different size & color) (3/4) (0)	2019.01.13
[Python] 그룹별 산점도 점 색깔과 모양 다르게 하기 (Scatter Plot by Groups) (2/4) (3)	2019.01.13
[Python] 산점도 그래프 (Scatter Plot) (1/4) (0)	2019.01.13
[Python] 원 그래프 (Pie Chart), 하위 그룹을 포함한 도넛 그래프 (Donut Chart with Subgroups) (0)	2019.01.12
[Python] 막대 그래프 (Bar Chart) (0)	2019.01.11