Runtimewaring: ongeldige waarde die is aangetroffen in deel

Ik moet een programma maken met behulp van de methode van Euler voor de “bal in een veer” -model

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()

Ik krijg deze foutmelding:

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

Ik kan het niet achterhalen, wat is er mis met de code?

Antwoord 1, Autoriteit 100%

Ik denk dat uw code probeert te “verdelen door nul” of “verdelen door NAN”. Als u zich hiervan bewust bent en niet wilt om u lastig te vallen, dan kunt u het proberen:

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

Zie voor meer informatie:

• http://docs.scipy.org/doc/ numpy / referentie / gegenereerd / numpy.seterr.html

Antwoord 2, Autoriteit 9%

Python-indexering begint bij 0 (in plaats van 1), dus uw opdracht “R [1,] = R0” definieert het tweede (dwz index 1) -element van R en laat de eerste ( Index 0) Element als een paar nullen. De eerste waarde van I in uw voor lus is 0, dus RR krijgt de vierkantswortel van het puntproduct van de eerste invoer in R met zichzelf (die 0) is, en de divisie door RR in de volgende lijn werpt de fout.

Antwoord 3, Autoriteit 7%

Om divisie op nul te voorkomen, kunt u de uitvoer vooraf initialiseren “waar de DIV0-fout optreedt, bijvoorbeeld np.whereKnipt het niet sinds de volledige lijn wordt geëvalueerd, ongeacht de conditie.

Voorbeeld met pre-initialisatie:

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

Antwoord 4, Autoriteit 2%

U scheidt met rrdie 0,0 kan zijn. Controleer of rrnul is en iets anders dan gebruikt dan het gebruiken in de noemer.