Построить линейное уравнение с использованием 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 при работе с numpy массивами.

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

Answer 2

, если вы хотите использовать numpy, то import numpy as np, если вы хотите использовать математику, тогда import math.

Answer 3

Вы уже импортировали numpy, проще использовать np.power вместо math.pow, так как он работает с массивами или даже просто с оператором **:

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()