python 足した配列の値の平均

Question

import numpy as np
import matplotlib.pyplot as plt
from numpy.lib.type_check import isrealobj

def f1(fai2):
    return fai2

def f2(fai1,fai2):
    return (i-np.sin(fai1)-fai2)/bc

a = 0.0
b = 100
N = 10000
h = (b-a)/N
j = 0
bc = 2.0
i = 1.2

t = np.arange(a,b,h)
fai1_0 = 0.0
fai2_0 = 0.0

fai1 = np.empty(N)
fai2 = np.empty(N)
fai1[0] = fai1_0
fai2[0] = fai2_0


#ルンゲクッタ法
for j in range (N-2):

    k1 = h * f1(fai2[j])
    d1 = h * f2(fai1[j] , fai2[j])

    k2 = h * f1(fai2[j] + d1/2)
    d2 = h * f2(fai1[j] + k1/2,  fai2[j] + d1/2)

    k3 = h * f1(fai2[j] + d2/2)
    d3 = h * f2(fai1[j] + k2/2 , fai2[j] +d2/2)

    k4 = h * f1(fai2[j] + d3)
    d4 = h * f2(fai1[j] + k3 , fai2[j] + d3)     

    fai1[j+1] = fai1[j] + 1/6 * (k1 + 2 * k2 + 2 * k3 + k4) 
    fai2[j+1] = fai2[j] + 1/6 * (d1 + 2 * d2 + 2 * d3 + d4) 

    if j == N-1:
      break

    j += 1
    
    #τが6000以降のグラフの値を足す　　　
    if j >= 6000:　　　　　　　　　　　　　　　
         sum(fai2[j])

#足した値を平均化する
v = sum / 4000

plt.scatter (1, v, marker='.')
plt.xlabel("τ")
plt.ylabel("φ2")
plt.show()
```前回に引き続いたコードで、ｔが6000から10000の時のfai2の値を合計して、その値を平均するコードをかきたいのですが、
```python
'numpy.float64' object is not iterable
  File "C:\Users\hot40\OneDrive\Desktop\rungekuttapra6.py", line 54, in <module>
    sum(fai2[j])
```とエラーが起きてしまいます。どう修正すればよいでしょうか。

Answer 1

numpy.average() を使います。平均値の計算は for ループが終わってから行います(fai2 に全ての値がセットされた後)。

    #τが6000以降のグラフの値を足す
    #if j >= 6000:
    #     sum(fai2[j])

#足した値を平均化する
#v = sum / 4000
v = np.average(fai2[6000:])

Comments

[Ackngawe.-]

できました！
いつも答えていただき、とても勉強になります！ありがとうございます！

[Ackngawe.-]

追加で一つお伺いしたいです！
v = np.average(fai2[6000:])というのは、ｊが０から6000のときのfai2の値を合計しているように思えるのですが、実際はどうなのでしょうか。

[melian]

ええとですね、j が 6000 以降の全ての fai2 の値の平均値を計算しています。 0 から 6000 の値を平均する場合は np.average(fai2[:600１]) とします。

[Ackngawe.-]

なるほど、、
：の位置でどちらかがきまるのですね！ありがとうございます！