[파이썬 matplotlib] 산점도 그룹별로 색 다르게 하기

산점도 그래프를 그릴 때 그룹별로 색을 구분하는 방법입니다. iris 데이터를 사용했습니다. 간단히 절차를 설명합니다. 코드 주석에도 설명이 적혀 있습니다. 

- iris 데이터를 불러옴
- iris 데이터를 데이터프레임으로 만듦
- groupby 메소드를 이용하여 데이터를 그룹별로 나눔
- for 문을 이용하여 그룹별로 색을 구분하여 산점도 그림

import pandas as pd
from sklearn.datasets import load_iris

#데이터 가져오기
iris=load_iris()

#데이터프레임으로 만들기
df=pd.DataFrame(iris.data,columns=iris.feature_names)
df['y']=iris.target
df['y']=df['y'].map({0:'setosa',1:'versicolour',2:'virginica'})


import matplotlib.pyplot as plt

#꽃 종류 별로 그룹화
groups = df.groupby('y')

#그래프 그리기
fig,ax=plt.subplots(1,1)
for name,data in groups:
    ax.plot(data['sepal length (cm)'], data['sepal width (cm)'], label=name,linestyle='',marker='o')
ax.legend()
plt.show()