python - 计算矩阵列平均值

Question

我有这个矩阵:

      [[[ 0.49757494  0.50242506]
      [ 0.50340754  0.49659246]
      [ 0.50785456  0.49214544]
         ..., 
      [ 0.50817149  0.49182851]
      [ 0.50658656  0.49341344]
      [ 0.49419885  0.50580115]]

      [[ 0.117       0.883     ]
      [ 0.604       0.396     ]
      [ 1.          0.        ]
         ..., 
      [ 0.98559675  0.01440325]
      [ 0.948       0.052     ]
      [ 0.012       0.988     ]]

      [[ 0.21099179  0.78900821]
      [ 0.75212493  0.24787507]
      [ 0.96653919  0.03346081]
           ..., 
      [ 0.97485074  0.02514926]
      [ 0.95051503  0.04948497]
      [ 0.05409603  0.94590397]]]

如果权重为 w1,w2,w3，如何计算每个矩阵(3 x 2)第一列和第二列的平均值？所以我可以得到这样的结果:

      [[[(X1        Y1]
        ...,
      [X2           Y2]
      [[X3          Y3]
        ...,

提前致谢。

编辑:输入形状是 (3, 37375, 2)，我想用 (1,2) 代替 (3,2)。我想获取每列的平均值，例如:

   (0.497*w1 + 0.503*w2 + 0.507*w3)/ (w1 + w2 + w3)     <--- First column

Answer 1

假设您的输入形状为 (3,n,2) 并且您希望形状为 (n,3,2) 您首先要执行的操作

in=in.reshape((-1,3,2))

如果你有一个权重向量w

w = np.random.rand(3)

然后您可以使用 np.average 对第一个轴进行加权平均(产生 (n,2)

out1 = np.average(in, weights = w, axis = 1)

或者你可以进行加权求和

out1 = np.sum(t*w[None,:, None], axis = 1) / np.sum(w)