RuntimeWarning：在除法中遇到无效的值

95

我必须使用欧拉方法为“弹簧中的球”模型编写程序

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

我无法弄清楚，代码有什么问题？

python python-2.7 matplotlib

— 博格丹·奥西卡（Bogdan Osyka）
source

打印该行代码中每个较小项目的内容。那是调试它的唯一方法。

— CppLearner

2

您有的nanfor rr，它抛出该错误。在某些情况下，的问题rr源于。正如@CppLearner提到的那样，调试（或编写）代码的最佳方法是在实施之前测试每个较小的部分。r[i,:]array([ nan, nan])

— 宇宙化

159

我认为您的代码正在尝试“除以零”或“除以NaN”。如果您知道并且不想让它困扰您，那么您可以尝试：

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

有关更多详细信息，请参见：

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

— 严竹
source

76

with NP.errstate(divide='ignore',invalid='ignore'):如果您要禁止显示代码块警告，则使用它很有用。

— GWW

8

为什么要忽略零或NaN除法？

— x

7

@xsquared当您自己正确地处理了值之后，在除法之后，您正在将代码分发给用户（或者不愿看到警告）。with np.errstate(...)让您仅在处理好的情况下安全地执行此操作。

— reve_etrange'3

2

@reve_etrange我发现比通常忽略零除要容易得多。

— x

1

最好在导致错误的行之前设置此值，然后在该行之后'warn'通过命令将其重置为正常状态np.seterr(divide='warn', invalid='warn')

— Mohammad ElNesr '19

15

Python索引从0（而不是1）开始，因此您的赋值“ r [1 ,:] = r0”定义r的第二个（即索引1）元素，而第一个（索引0）元素保留为一对零。for循环中i的第一个值为0，因此rr获取r中第一个条目与自身的点积的平方根（即0），并且在下一行中用rr除以引发错误。

— 绞车
source

10

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

— 质量标准
source

4

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

— 蜡ze
source