[Numpy] Poly1d 그래프 plot

[Numpy] 다항식을 다룰 수 있는 polynomial class, poly1d

지난 번에 다항식을 다룰 수 있도록 해주는 numpy의 poly1d class에 대해 알아보았습니다.

stack overflow에서 polynomial을 plot하는 내용의 질문을 보고 한 번 정리해두려 합니다.

 

import matplotlib.pyplot as plt
import numpy as np

p = poly1d([1, 2, 1])
plt.plot(p)

위 다항식을 poly1d 클래스로 만들어서 plot하면 결과는 어떻게 될까요?

원래 생각했던 그래프와 다른 엉뚱한 그래프가 나왔습니다. matplotlib의 plot함수는 x와 y의 값이 같이 주어져야하는데 주어진 다항식의 계수 [1, 2, 1]로만 그래프가 만들어져서 이런 결과가 나왔습니다.

X축의 범위를 같이 넣어주면 원하는 그래프를 얻을 수 있습니다.

import matplotlib.pyplot as plt
import numpy as np

p = np.poly1d([1, 2, 1])
X = np.arange(-4, 2.1, 0.1)
y = p(X)
plt.plot(X, y)