RuntimeWarning: napotkano nieprawidłową wartość podczas dzielenia

101

Muszę zrobić program metodą Eulera dla modelu „kulka na sprężynie”

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

Ciągle otrzymuję ten błąd:

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

Nie mogę tego rozgryźć, co jest nie tak z kodem?

Bogdan Osyka
źródło
wypisuje, co się dzieje w każdym z mniejszych elementów w tym wierszu kodu. To jedyny sposób na debugowanie.
CppLearner
3
Masz nans dla rr, który rzuca ten błąd. Problem rrwynika z r[i,:]tego, z czego w niektórych przypadkach równa się array([ nan, nan]). Jak wspomniał @CppLearner, najlepszym sposobem debugowania (lub pisania) kodu jest przetestowanie każdej mniejszej części przed wdrożeniem.
kosmoza

Odpowiedzi:

165

Myślę, że twój kod próbuje „podzielić przez zero” lub „podzielić przez NaN”. Jeśli jesteś tego świadomy i nie chcesz, żeby Ci to przeszkadzało, możesz spróbować:

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

Więcej informacji:

Yan Zhu
źródło
77
Może być przydatny, with NP.errstate(divide='ignore',invalid='ignore'):jeśli chcesz pominąć ostrzeżenia dla bloku kodu.
GWW
8
Dlaczego ktoś miałby chcieć zignorować dzielenie przez zero lub NaN?
x do kwadratu
7
@xsquared Gdy sam poprawnie obsłużyłeś wartość, po podziale i rozprowadzasz swój kod wśród użytkowników (lub masz dość oglądania ostrzeżenia). with np.errstate(...)pozwala to zrobić bezpiecznie tylko dla obsługiwanej obudowy.
reve_etrange
2
@reve_etrange To, że uważam za dużo bardziej akceptowalne niż generalne ignorowanie dzielenia przez zero.
x do kwadratu
1
lepiej ustawić to przed linią, która powoduje błąd, a następnie zresetować po linii do normalnego stanu 'warn'za pomocą polecenianp.seterr(divide='warn', invalid='warn')
Mohammad ElNesr
15

Indeksowanie Pythona zaczyna się od 0 (zamiast 1), więc twoje przypisanie „r [1 ,:] = r0” definiuje drugi (tj. Indeks 1) element r i pozostawia pierwszy element (indeks 0) jako parę zer. Pierwsza wartość i w twojej pętli for to 0, więc rr pobiera pierwiastek kwadratowy z iloczynu skalarnego pierwszego wpisu w r ze sobą (czyli 0), a dzielenie przez rr w kolejnym wierszu generuje błąd.

Kinch
źródło
11

Aby zapobiec dzieleniu przez zero, możesz wstępnie zainicjować wyjście 'out', w którym występuje błąd div0, np. np.whereNie powoduje to jego wycięcia, ponieważ cała linia jest oceniana niezależnie od warunku.

przykład z pre-inicjalizacją:

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
qrtLs
źródło
4

Dzielisz, przez rrktóry może być 0,0. Sprawdź, czy rrjest zero i zrób coś rozsądnego poza używaniem go w mianowniku.

crayzeewulf
źródło