如何取数组中各个元素的平均值

Question

我有一段代码运行 1000 次并生成 1000 个协方差矩阵。如何计算矩阵中每个元素的平均值，然后打印该平均矩阵？

params_avg1=[]
pcov1avg=[]
i=1000
for n in range(i):
    y3=y2+np.random.normal(loc=0.0,scale=.1*y2)
    popt1,pcov1=optimize.curve_fit(fluxmeasureMW,bands,y3)
    params_avg1.append(popt1)
    pcov1avg.append(pcov1) #returns an array of 1000 3x3 covariance matrices

Answer 1

由于您已经将所有矩阵附加到一个数组中，因此将其转换为 3D numpy 数组，然后在正确的轴上进行平均：

np.array(pcov1avg).mean(axis=0) # or equivalently np.mean(pcov1avg, 0) 

关于命名的一点点 - i 通常表示迭代的当前索引而不是结束值，通常表示为 n