python - 如何根据列表的长度更改取平均值？ (Python)

Question

我有一个列表，其中包含包含数组的列表，如下所示:

A = [
    [
        array([[ 1.,  4.3, 0.,  0.],
               [ 0.,  0.,  0.,  0.],
               [ 0.,  0.,  0.,  0.],
               [ 0.,  0.,  0.,  0.]])
    ],

    [
        array([[ 5.,  0.,  0.,  0.],
               [ 0.,  0.,  0.,  0.],
               [ 0.,  0.,  0.,  0.],
               [ 0.,  0.,  0.,  0.]])
    ]
]       

所以我基本上在这个东西里面有两个 4x4 矩阵。现在，目标是取这两个我设法做到的平均值:

np.mean([A[0][0],A[1][0]],axis=0)

我还有另一个矩阵 B，它由三个 4x4 矩阵组成，然后平均值将类似于

np.mean([B[0][0],B[1][0]],B[2][0],axis=0)

我想对此进行概括，这样我就不必每次都重写 np.mean 部分。所以我可能会使用 A (2) 的长度或 B (3) 的长度来构造它，但我不确定如何获得类似的东西

np.mean(C[0][0],C[1][0],[...][0],[n-1][0],axis=0)

其中 n 是 len(C)。

我该如何实现？谢谢!

Answer 1

你可以只使用列表理解:

>>> np.mean([A[i][0] for i in range(len(A))], axis=0)

或更短、更具可读性和“pythonic”:

>>> np.mean([a[0] for a in A], axis=0)
array([[ 3.  ,  2.15,  0.  ,  0.  ],
       [ 0.  ,  0.  ,  0.  ,  0.  ],
       [ 0.  ,  0.  ,  0.  ,  0.  ],
       [ 0.  ,  0.  ,  0.  ,  0.  ]])