점의 크기 조절하기
다른 그래프는 항목을 2개까지 표현할 수 있지만, 산점도는 세 항목도 한 번에 표현할 수 있습니다. x축, y축으로 두 항목을, 점의 크기로 세 번째 항목을 표현합니다. 점의 크기는 plt.scatter()에 s 옵션으로 추가해 조절합니다.
키와 몸무게가 체육 점수에 영향을 미치는지 알아봅시다. 이를 위해 키와 몸무게, 체육 점수의 상관관계를 살펴봅시다. 앞의 코드에 체육 점수를 의미하는 score 리스트를 추가하고 점의 크기를 score로 지정합니다. 체육 점수는 점의 크기가 뚜렷이 차이나도록 1부터 1000 사이의 무작위 값을 넣겠습니다.
height, weight, score = [], [], []
for i in range(100):
height.append(random.randint(100, 200))
weight.append(random.randint(20, 100))
score.append(random.randint(1, 1000))
plt.rc('font', family='Malgun Gothic')
plt.title('키와 몸무게, 체육 점수의 상관관계')
plt.scatter(height, weight, s=score, c=score, cmap='rainbow', alpha=0.7)
plt.xlabel('키')
plt.ylabel('몸무게')
plt.colorbar(label='체육 점수')
plt.show()