여기에 연령에 따른 컬러맵을 적용합니다. 또 남성 인구수 중 가장 큰 값을 기준으로 y = x 형태의 직선, 즉 추세선을 그려 어떤 성별의 인구가 더 많은지 한눈에 들어오도록 합니다.

import matplotlib.pyplot as plt
plt.scatter(m, f, c= range(101), alpha=0.5, cmap='jet') # 컬러맵 적용
plt.colorbar()
plt.plot(range(max(m)),range(max(m)), 'g') # 추세선 추가
plt.show()

그림 10-16 연령을 컬러맵으로 표현하고 추세선을 추가한 결과

마지막으로 연령별 남성과 여성 인구수를 합친 값을 size 리스트에 넣어 점의 크기를 표현하면 완성입니다. 이때 math 라이브러리에 있는 제곱근 함수인 sqrt()를 사용하여 점의 크기를 적절히 조절하고 x, y축의 이름을 xlabel(), ylabel() 함수로 정해줍니다.

다음은 이를 적용한 전체 코드입니다.