运行时警告: 分割时遇到的无效值

我必须使用欧拉的方法为“弹簧中的球”模型制作一个程序

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

我不明白，密码有什么问题？

374921 次浏览

小开

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

小开

最佳答案

我认为你的代码是试图“除以零”或“除以 NaN”。如果你意识到了这一点，并且不希望它打扰到你，那么你可以尝试:

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

详情请参阅:

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

小开

Python 索引从0(而不是1)开始，因此赋值“ r [1，: ] = r0”定义 r 的第二(即索引1)元素，并将第一个(索引0)元素保留为一对零。For 循环中 i 的第一个值是0，因此 rr 得到 r 中第一个条目的点乘的平方根(也就是0) ，后面一行中除以 rr 将抛出错误。

小开

为了防止除以零，您可以在 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