代码如下:
a = np.array([[1, 4, 2, 6],
[10, 41, 7, 3],
[9, 1, 6, 2]])
v1 = np.average(a, weights=[3, 3, 4], axis=0)
print(v1)
运行结果
当执行这段代码时,np.average(a, weights=[3, 3, 4], axis=0)
会根据指定的权重在列方向上计算加权平均值。
具体计算过程如下:
对于第一列:
对于第二列:
对于第三列:
对于第四列:
最终输出的结果 v1
为: