[데이터분석] matplotlib : 산점도 그래프 그리기 - scatter

matplotlib 라이브러리로 산점도 그래프도 그릴 수 있는데요, matplotlib.pyplot 에서 제공하는 scatter() 함수를 이용하여 그릴 수 있습니다.

 

 

먼저, x축 데이터와 y축 데이터는 랜덤으로 난수를 100개씩 생성하겠습니다.

 

import matplotlib.pyplot as plt
import numpy as np

np.random.seed(0)

N = 100
x = np.random.rand(N)
y = np.random.rand(N)

 

한편, scatter() 함수를 이용하여 산점도 그래프를 그리기 전, 색깔과 산점도 영역을 랜덤으로 지정해주어야 하는데요, 다음과 같이 입력하면 됩니다.

 

colors = np.random.rand(N)
area = (30 * np.random.rand(N))**2

 

각 파라미터 값은 출력된 그래프를 보고 조금씩 수정하시면 됩니다.

 

위 코드를 실행했다면 scatter() 함수를 이용하여 다음과 같이 코드를 입력하면 되는데요,

scatter() 함수 안에 인자로 x축 데이터, y축 데이터, 지정한 영역 값, 색깔, 투명도를 차례로 넣으면 됩니다.

 

plt.scatter(x, y, s=area, c=colors, alpha=0.5)
plt.show()

 

출력된 그래프를 보면 산점도 그래프로 잘 표현되었음을 확인할 수 있습니다.