[Python] Numpy(넘파이)로 그래프 그리기

Numpy와 matplotlib을 활용하여 여러 가지 그래프 그리기

이전 시간 Numpy를 활용하여 리스트를 만들어 보았습니다. 이번에는 Numpy를 활용하여 데이터 분석에 필요한 여러 종류의 그래프를 그려보려 합니다.

그래프를 그리기 위해서는 matplotlib 모듈에 있는 pyplot을 불러와야 합니다.

import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline

pyplot은 plt로 줄여 부르며, %matplotlib inline을 통하여 jupyter notebook에 표현됩니다.

지난 시간 배운 numpy random을 활용하면,(참고: [Python] Numpy(넘파이)로 행렬 만들기)

rand_x = np.arange(30)
rand_y = np.random.randn(30)

여기서 randn은 '평균값'은 0이며, '분산'이 1인 값을 추출하는 수식입니다.

• 분산 정의

rand_y에 30개를 랜덤으로 추출하면,

array([-0.37010144, 2.50793966, -0.4419649 , 0.48033018, -1.07036293, 0.80167091, 2.03569028, -1.8043962 , 0.76847793, 0.14383409, 1.24775824, 0.22388488, -0.24057212, 2.29708151, -0.75715921, -0.31227642, -0.67988752, -0.59097732, 0.58303298, -1.31041021, 0.7259335 , -0.29697225, -0.2736434 , -0.56686133, 1.01358154, -0.39807358, 0.69855798, -0.06129629, 0.05048929, 0.26471836])

이 출력됩니다. 위 값을 이용하여 그래프를 만들어 보겠습니다.

---

1. Plot 그래프

plt.plot(rand_x, rand_y) 
plt.show()

 

2. Bar 그래프

plt.bar(rand_x,rand_y) 
plt.show()

3. Scatter

plt.scatter(rand_x,rand_y) 
plt.show()

4. Barh

plt.barh(rand_x,rand_y) 
plt.show()

5. fill_between

plt.fill_between(rand_x,rand_y) 
plt.show()

참고

• https://matplotlib.org/gallery/index.html

---

Tip. 유용한 선 옵션

색 추가

색 종류	의미
'b'	파랑
'g'	초록
'r'	빨강
'c'	청록
'm'	자주
'y'	노랑
'k'	검정
'w'	흰색

#선 색깔 빨강 plt.plot(rand_x,rand_y,'r') plt.show()

선 종류

선 종류	의미
"."	점
","	쉼표
"o"	원
"v"	아래 삼각형
"^"	위 삼각형
"<"	왼쪽 삼각형
">"	오른쪽 삼각형
"1"	아래 삼각뿔
"2"	위 삼각뿔
"3"	왼쪽 삼각뿔
"4"	오른쪽 삼각뿔
"8"	팔각형
"s"	사각형
"*"	별 모양
"+"	십자가
"x"	x 모양
"D"	다이아몬드
"d"	얇은 다이아몬드

#빨간선 and 별 추가 plt.plot(rand_x, rand_y, 'r*') plt.show()

좌) 빨강 + 별 추가 // 우) 기존 scatter 그래프