Question

我必须使用Euler的方法为“春天的球”模型制作一个程序

from pylab import*
from math import*
m=0.1
Lo=1
tt=30
k=200
t=20
g=9.81
dt=0.01
n=int((ceil(t/dt)))
km=k/m
r0=[-5,5*sqrt(3)]
v0=[-5,5*sqrt(3)]
a=zeros((n,2))
r=zeros((n,2))
v=zeros((n,2))
t=zeros((n,2))
r[1,:]=r0
v[1,:]=v0
for i in range(n-1):
    rr=dot(r[i,:],r[i,:])**0.5
    a=-g+km*cos(tt)*(rr-L0)*r[i,:]/rr
    v[i+1,:]=v[i,:]+a*dt
    r[i+1,:]=r[i,:]+v[i+1,:]*dt
    t[i+1]=t[i]+dt

    #print norm(r[i,:])

plot(r[:,0],r[:,1])
xlim(-100,100)
ylim(-100,100)
xlabel('x [m]')
ylabel('y [m]')

show()

我一直收到这个错误：

a=-g+km*cos(tt)*(rr-L0)*r[i,:]/rr
RuntimeWarning: invalid value encountered in divide

我无法弄明白，代码出了什么问题？

Answer 1

我认为您的代码正在尝试“除以零”＃34;或者＆＃34;除以NaN＆＃34;。如果您知道这一点并且不想让它打扰您，那么您可以尝试：

import numpy as np
np.seterr(divide='ignore', invalid='ignore')

有关详细信息，请参阅：

http://docs.scipy.org/doc/numpy/reference/generated/numpy.seterr.html

Answer 2

Python索引从0（而不是1）开始，因此你的赋值“r [1，：] = r0”定义r的 second （即索引1）元素并离开第一个（ index 0）元素作为一对零。你的for循环中i的第一个值是0，所以rr得到r中第一个条目的点积的平方根（它是0），并且后续行中rr的除法会抛出错误。 / p>

Answer 3

您除以rr，可能是0.0。检查rr是否为零并做一些合理的事情，而不是在分母中使用它。

Answer 4

为防止被零除，您可以在div0错误发生的地方预初始化输出“ out”，例如np.where不会削减它，因为无论条件如何，整个行都会被评估。

具有预初始化的示例：

a = np.arange(10).reshape(2,5)
a[1,3] = 0
print(a)    #[[0 1 2 3 4], [5 6 7 0 9]]
a[0]/a[1]   # errors at 3/0
out = np.ones( (5) )  #preinit
np.divide(a[0],a[1], out=out, where=a[1]!=0) #only divide nonzeros else 1

RuntimeWarning：在divide中遇到无效值

4 个答案: