Использование уравнений Sympy для построения графика
Каков наилучший способ создания уравнения Симри, сделать что-то вроде взятия производной, а затем построить результаты этого уравнения?
У меня есть мое символическое уравнение, но не могу понять, как создать массив значений для построения. Здесь мой код:
from sympy import symbols
import matplotlib.pyplot as mpl
t = symbols('t')
x = 0.05*t + 0.2/((t - 5)**2 + 2)
nums = []
for i in range(1000):
nums.append(t)
t += 0.02
plotted = [x for t in nums]
mpl.plot(plotted)
mpl.ylabel("Speed")
mpl.show()
В моем случае я просто вычислил производную этого уравнения, и теперь я хочу построить скорость x
, поэтому это довольно упрощено.
Ответы
Ответ 1
Вы можете использовать numpy.linspace()
для создания значений оси x (x_vals
в коде ниже) и lambdify()
.
from sympy import symbols
from numpy import linspace
from sympy import lambdify
import matplotlib.pyplot as mpl
t = symbols('t')
x = 0.05*t + 0.2/((t - 5)**2 + 2)
lam_x = lambdify(t, x, modules=['numpy'])
x_vals = linspace(0, 10, 100)
y_vals = lam_x(x_vals)
mpl.plot(x_vals, y_vals)
mpl.ylabel("Speed")
mpl.show()
(улучшения, предложенные asmeurer и MaxNoe)
![введите описание изображения здесь]()
В качестве альтернативы вы можете использовать sympy plot()
:
from sympy import symbols
from sympy import plot
t = symbols('t')
x = 0.05*t + 0.2/((t - 5)**2 + 2)
plot(x, (t, 0, 10), ylabel='Speed')
Ответ 2
Использование SymPy
Вы можете напрямую использовать функции построения графиков SymPy:
from sympy import symbols
from sympy.plotting import plot as symplot
t = symbols('t')
x = 0.05*t + 0.2/((t - 5)**2 + 2)
symplot(x)
![enter image description here]()
Большую часть времени он использует matplotlib в качестве бэкэнда.