컬러맵 적용하기

이번에는 데이터를 바꿔 키와 몸무게의 상관관계를 표현하는 산점도를 그려 보겠습니다. 100명의 데이터를 표시하는데, 숫자 100개를 직접 입력하긴 어려우므로 random 라이브러리를 사용해 무작위로 키와 몸무게 100개를 만듭니다. 키를 저장하는 리스트는 height, 몸무게를 저장하는 리스트는 weight로 하고, 반복문을 활용해 각 리스트에 숫자 100개를 넣습니다. 그리고 컬러맵(color map)을 사용해 산점도를 다채롭게 표현합니다. 컬러맵을 사용하려면 2가지 옵션을 추가해야 합니다. cmap 옵션으로 사용할 컬러맵의 이름을 지정하고, c 옵션으로 컬러맵을 표현할 기준이 되는 축을 설정합니다. 코드를 봅시다.

import random

height, weight = [], []
for i in range(100):
    height.append(random.randint(100, 200))
    weight.append(random.randint(20, 100))
plt.rc('font', family='Malgun Gothic')
plt.title('키와 몸무게의 상관관계')
plt.scatter(height, weight, c=height, cmap='RdPu')
plt.xlabel('키')
plt.ylabel('몸무게')
plt.show()

실행결과