[Python] 파이썬을 이용하여 함수 그래프 그리기 (NumPy, matplotlib.pyplot)

파이썬을 이용하여 함수 그래프 그리기 (NumPy, matplotlib.pyplot)

개요

• 파이썬의 넘파이(NumPy)와 matplotlib.pyplot을 사용하여 간단하게 그래프를 그릴 수 있다.

 

방법

코드

import numpy as np
import matplotlib.pyplot as plt

# (1) X축의 범위(Range) 설정 (입력)
start = -10         # 시작점           # 입력
end = 10            # 끝점 + 1         # 입력
interval = 1        # 간격               # 입력

x = np.array(range(start, end, interval))
# print('X: ', x)    # X 축의 범위를 확인하고 싶을 경우 주석 해제

# (2) 그래프의 축 이름 설정 
name_X = 'Axis-X'     # 입력
name_Y = 'Axis-Y'     # 입력

plt.xlabel(name_X)
plt.ylabel(name_Y)

# (3) 함수 추가 (입력)
## 1차식
eq_1 = 30*x + 10                     
lbl_1 = 'y = ' + '30*x + 10' 

## 2차식
eq_2 = 20*x**2 -3*x - 15
lbl_2 = 'y = ' + '20*x**2 -3*x - 15'

## 3차식
eq_3 = x**3 -2*x**2 - 2*x + 3
lbl_3 = 'y = ' + 'x**3 -2*x**2 - 2*x + 3'

## ...(추가)...

# (4) 그리드 추가하기
color_grid = "gray"    # 입력
alpha_grid = 0.5        # 입력
linestyle_grid = '--'    # 입력

plt.grid(color=color_grid, alpha=alpha_grid, linestyle=linestyle_grid)

# (5) 그래프 표현하기
plt.plot(x, eq_1, label=lbl_1)
plt.plot(x, eq_2, label=lbl_2)
plt.plot(x, eq_3, label=lbl_3)

# (6) 범례 작성
plt.legend()
plt.show()

 

• 코드에서 '입력' 주석이 달린 부분을 표현하고자 하는 함수의 조건에 맞게 수정해주면 된다.

 

---

참고 : https://matplotlib.org/3.5.0/api/_as_gen/matplotlib.pyplot.plot.html