使用matplotlib绘制线性方程

Question

我正在尝试使用matplotlib绘制方程式。如果我输入了一些虚拟数据：

from numpy import arange
i = arange(0.0, 50.0)

并按如下所示插入方程式：

y4 = .00001              
y3 = .00002            
y2 = .00003               
y1 = .00004              
y0 = .00005

x = y4 * math.pow(i, 4) + y3 * math.pow(i, 3) + y2 * math.pow(i, 2) + y1 * i + y0             

它返回错误：

TypeError: only length-1 arrays can be converted to Python scalars

Answer 1

您已经导入了numpy，使用np.power而不是math.pow会更容易，因为它可以处理数组，甚至可以使用**运算符：< / p>

x = y4 * np.power(i, 4) + y3 * np.power(i, 3) + y2 * np.power(i, 2) + y1 * i + y0

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

Answer 2

在使用numpy数组时，应该使用numpy。

x = y4 * i**4 + y3 * i**3 + y2 * i**2 + y1 * i + y0

Answer 3

如果要使用numpy，请使用import numpy as np，如果要使用数学，则请使用import math。