web-dev-qa-db-ja.com

RuntimeWarning：除算で無効な値が見つかりました

「春のボール」モデルにオイラーの方法を使用してプログラムを作成する必要があります

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

理解できませんが、コードの何が問題になっていますか？

pythonpython-2.7matplotlib
62
Bogdan Osyka

あなたのコードは「ゼロで除算」または「NaNで除算」しようとしていると思います。あなたがそれを知っていて、それがあなたを煩わせたくないなら、あなたは試すことができます：

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

詳細については、以下を参照してください。

106
Yan Zhu

Pythonのインデックス作成は0（1ではなく）で始まるため、割り当て "r [1 ,:] = r0"はrのsecond（つまりインデックス1）要素を定義し、最初の（インデックス0）を残します。ゼロのペアとしての要素。 forループのiの最初の値は0であるため、rrはrの最初のエントリとそれ自体（0）のドット積の平方根を取得し、後続の行のrrによる除算はエラーをスローします。

12
Kinch

rrで割っています。0.0の場合があります。 rrがゼロであるかどうかを確認し、分母で使用する以外の合理的な操作を行います。

4
crayzeewulf

ゼロによる除算を防ぐために、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
2
qrtLs