밑바닥부터 시작하는 데이터 과학 데이터 분석을 위한 파이썬 프로그래밍과 수학·통계 기초

Data Science

296

󻱴󽴔󽴔󺈿󻺠󼬣

20.5

예시: 색 군집화하기

굿즈 팀 부사장은 오프라인 모임에서 제공할 수 있는 데이텀 스티커를 여러 장

제작했다. 그런데 아쉽게도 스티커 프린터는 총 다섯 가지 색밖에 출력할 수가

없다. 설상가상으로 디자인 팀 부사장이 휴가를 가버리는 바람에, 굿즈 팀 부사

장은 어떻게든 디자인을 다섯 가지 색만 사용하도록 바꿔줄 수 없냐고 묻는다.

이미지 파일은 보통 픽셀의

2

차원 배열이고, 픽셀은 색을 나타나는

(red,

green, blue)

로 구성된

3

차원 벡터이다.

이것을 다섯 가지 색의 이미지로 바꾼다는 것은 다음의 과정을 의미한다.

1

. 다섯 가지 색을 고른다.

2

. 각 픽셀을 다섯 가지 색 중 하나로 매핑한다.

결국 이 작업도

k

-

means

로 풀기에 적당한 형태이다.

k

-

means

를 사용하면

RGB

공간상에서 모든 픽셀을 다섯 개의 군집으로 나눌 수 있다. 그 다음에 각 픽셀을

각 군집의 중점색으로 칠하면 된다.

먼저, 파이썬으로 이미지를 읽어 보자.

pillow

라이브러리를 설치하고 나면

mat

-

plotlib

로 쉽게 할 수 있다.

python -m pip install pillow

다음으로

matplotlib.image.imread

를 사용하면 된다.

image

_

path = r"girl

_

with

_

book.jpg" # ...