Runge Kutta Not Plotを計算する関数

Question

こんにちは私はRunge Kuttaメソッドを使用してODEを解決しプロットするスクリプトに取り組んでいます。私はスクリプトが異なる機能を使用するようにしたいので、後でそれを拡張することができます。関数定義を書いてもうまく動作しますが、def（）を使うとプロットウィンドウを開くことも、結果を出力することもありません。あなたより！

from numpy import * 
import matplotlib.pyplot as plt 

#H=p^2/2-cosq 
#p=dp=-dH/dq 
#q=dq=dH/dp 

t = 0 
h = 0.5 
pfa = []      #Create arrays that will hold pf,qf values 
qfa = [] 

while t < 10: 
    q = 1*t 
    p = -sin(q*t) 

    p1 = p 
    q1 = q 
    p2 = p + h/2*q1 
    q2 = q + h/2*p1 
    p3 = p+ h/2*q2 
    q3 = q+ h/2*p2 
    p4 = p+ h/2*q3 
    q4 = q+ h/2*p4 
    pf = (p +(h/6.0)*(p1+2*p2+3*p3+p4)) 
    qf = (q +(h/6.0)*(q1+2*q2+3*q3+q4)) 

    pfa.append(pf)     #append arrays 
    qfa.append(qf) 
    t += h       #increase time step       

print("test") 
plt.plot(pfa,qfa) 
print("test1") 
plt.show() 
print("tes2t") 

Answer 1

あなたは関数が宣言されている場合は、例えば、いくつかの点で、それを呼び出す必要があります：

def rk(p,q,h): 
    pass # your code here 

if __name__ == '__main__': 
    rk(1,2,1) 

がif __name__ == '__main__'ブロック内の関数呼び出しを置く機能がのみ呼び出されることを保証しますスクリプトを直接実行すると、別のスクリプトからインポートするときとは異なります。 （もっと興味がある場合は、こちらをご覧ください：What does if __name__ == "__main__": do?）

さらに良いオプションがあります。

def rk(p,q,h): 
    pass # your code here 

if __name__ == '__main__': 
    import argparse 
    the_parser = argparse.ArgumentParser() 
    the_parser.add_argument('integers', type=int, nargs=3) 
    args = the_parser.parse_args() 
    p,q,h = args.integers 
    rk(p,q,h)