Я не могу решить проблему "ось -1 выходит за границы массива измерения 0" - PullRequest
Новая
       48

Я не могу решить проблему "ось -1 выходит за границы массива измерения 0"

0 голосов
/ 13 декабря 2018

Я пытаюсь смоделировать движение пружинного маятника, и это мой код: 

import numpy as np
from scipy.integrate import odeint
from numpy import sin, cos, pi, array
import matplotlib.pyplot as plt


#Specify initial conditions
init = array([pi/18, 0]) # initial values

def deriv(z, t):
    x,y=z
    dy=np.diff(y,1)
    dy2=np.diff(y,2)
    dx=np.diff(x,1)
    dx2=np.diff(x,2)
    dt=np.diff(t,1)
    dt2=np.diff(t,1)
    dx2dt2=(4+x)*(dydt)^2-5*x+9.81*cos(y)
    dy2dt2=(-9.81*sin(y)-2*(dxdt)*(dydt))/(l+x)
    return np.array([dx2dt2,dy2dt2])

time = np.linspace(0.0,10.0,1000)
y = odeint(deriv,init,time)

plt.xlabel("time")
plt.ylabel("y")
plt.plot(time, y)
plt.show()

Я получаю сообщение об ошибке 

Traceback (most recent call last):
  File "/Users/cnoxon/Desktop/GRRR.py", line 24, in <module>
    y = odeint(deriv,init,time)
  File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/scipy/integrate/odepack.py", line 233, in odeint
    int(bool(tfirst)))
  File "/Users/cnoxon/Desktop/GRRR.py", line 13, in deriv
    dy=np.diff(y,1)
  File "/Library/Frameworks/Python.framework/Versions/3.7/lib/python3.7/site-packages/numpy/lib/function_base.py", line 1163, in diff
    axis = normalize_axis_index(axis, nd)
numpy.core._internal.AxisError: axis -1 is out of bounds for array of dimension 0

Я начинающийв Python, поэтому я не буду понимать большую часть терминологии, поэтому, пожалуйста, потерпите меня.Как мне исправить эту проблему?Я пытаюсь построить решения для двух уравнений 

dx2dt2=(4+x)*(dydt)^2-5*x+9.81*cos(y)
dy2dt2=(-9.81*sin(y)-2*(dxdt)*(dydt))/(l+x)

, но у меня много проблем.Кто-нибудь, пожалуйста, объясните мне, как мне переписать свой код, чтобы решить эту проблему?

Спасибо!

1 Ответ

0 голосов
/ 13 декабря 2018

Проблема возникает потому, что x и y являются целыми числами, а не массивами, поэтому вы не можете сделать np.diff(y,1).

Но ваша проблема глубже.Каждая запись в массиве y должна полностью описывать вашу систему, это означает, что каждое значение, необходимое для вычисления dx2dt2 и dy2dt2, должно находиться в этом векторе.Таким образом, y должен быть списком [x, y, dxdt, dydt].(Адаптируйте init, чтобы соответствовать этому)

Тогда ваша deriv функция просто должна дать производную от такого вектора, который равен: [dxdt, dydt, dx2dt2, dy2dt2].Ваша deriv функция становится очень простой! 

def deriv(z, t):
    x, y, dxdt, dydt = z

    dx2dt2=(4+x)*(dydt)^2-5*x+9.81*cos(y)
    dy2dt2=(-9.81*sin(y)-2*(dxdt)*(dydt))/(l+x)

    return np.array([dxdt, dydt, dx2dt2, dy2dt2])

И у вас есть еще две маленькие ошибки: используйте ** вместо ^ в python, и я думаю, что вы изменили 1в l ...

...