02. [Python] matplotlib 기초 : linspace함수와 그래프 그리기

개요

matplotlib로 그래프를 그릴 때에는 NumPy의 linspace()함수를 사용합니다.
`linspace()' 함수의 사용법과 그래프 그리는 방법에 대해서 알아보도록 하겠습니다.

실습환경

실습환경은 아나콘다와 Jupyter Notebook을 사용합니다.
아나콘다 설치 및 사용방법은 아래 링크를 확인해 주세요.
https://iot-lab.tistory.com/113

파이썬 아나콘다(Anaconda) 설치하기

linspace함수로 등간격의 배열 만들기

linespace()함수는 두 수 사이를 50개의 균일한 간격의 수를 배열로 만들어준다.
이 배열은 그래프의 Y축으로 많이 사용합니다.

import numpy as np

x = np.linspace(-5, 5)      # -10에서 10까지 50으로 등간격 배열 생성

print(x)
print('----------------------------')
print('x의 요소 개소 :', len(x))   # x의 요소 개수

[결과]

그래프 그리기

plplot를 사용해서 직선을 그릴 때에는 NumPy의 linspace()함수를 이용해서 x좌표 데이터를 배열로 생성하고, 각 x좌표에 값을 곱셈으로 y좌표의 데이터를 배열로 만듭니다.
그리고 pyplot의 plot()함수와 show()함수로 그래프를 표시합니다.

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(-10, 10)        # -10에서 10까지 50으로 등간격 배열 생성
y = x * 2                       # x에 2를 곱해서 y좌표를 생성

plt.plot(x, y)
plt.show()

 

-> 혹시 실행해도 그래프가 안 나오는 경우에는 %matplotlib inline 제일 상단에 추가해보시기 바랍니다.

 

 

다음에는 그래프를 꾸미는 방법에 대해서 알아보도록 하겠습니다.