8.3.3 산점도 겹쳐 그리기
이번에는 산점도 2개를 겹쳐서 표현해 보겠습니다. 키와 몸무게의 상관관계를 그리는데, 이번에는 그룹1과 그룹2에 각 100명씩 있다고 가정합니다. 그룹1의 키와 몸무게, 그룹2의 키와 몸무게를 산점도에 각기 다른 색상으로 표시해 봅시다.
그룹1의 키와 몸무게는 height1, weight1 리스트에, 그룹2의 키와 몸무게는 height2, weight2 리스트에 저장합니다. 예시이므로 총 400개의 값을 random 라이브러리를 사용해 무작위로 넣습니다. 비교를 위해 그룹1과 그룹2의 키와 몸무게 범위를 다르게 설정합니다. 그룹1의 키는 140~180, 그룹2의 키는 160~200으로, 그룹1의 몸무게는 40~80, 그룹2의 몸무게는 50~100으로 설정합니다.
import random
height1, weight1, height2, weight2 = [], [], [], []
for i in range(100):
height1.append(random.randint(140, 180))
weight1.append(random.randint(40, 80))
height2.append(random.randint(160, 200))
weight2.append(random.randint(50, 100))
plt.rc('font', family='Malgun Gothic')
plt.title('키와 몸무게의 상관관계')
plt.scatter(height1, weight1, color='crimson', alpha=0.7, label='그룹1')
plt.scatter(height2, weight2, color='indigo', alpha=0.7, label='그룹2')
plt.xlabel('키')
plt.ylabel('몸무게')
plt.show()
실행결과
