n x k阵列的平均值和RMSE

Question

我有这个目标数组：

[ 0.88  0.51  0.55  0.59  0.7 ]

和此示例数组：

[[ 0.4   0.02  0.52  0.44  0.48]
 [ 0.53  0.73  0.13  0.15  0.78]
 [ 0.67  0.27  0.26  0.31  0.17]
 [ 0.37  0.51  0.98  0.2   0.57]]

我想生成另一个计算

的数组（比如说＆＃39; fns＆＃39;）

• row0：样本数组的每列的平均值
• row1：每列的平均值+1标准偏差
• row2：每列的平均值-1标准偏差
• 第3行：每列平均值与平均值的RMSE

任何人都可以提出比嵌套语句更好的建议吗？

Answer 1

您可以使用许多numpy方法可用的axis参数来避免嵌套for循环。

fns = np.empty((4,sample.shape[1]))
stdv = np.std(sample,axis=0)
fns[0,:] = np.mean(sample,axis=0)
fns[1,:] = fns[0,:] - stdv
fns[2,:] = fns[0,:] + stdv
fns[3,:] = np.sqrt(np.mean((sample - target)**2,axis=0))