numpy数组中每个矩阵的轨迹

Question

给出多个矩阵的列表，我想计算每个矩阵的踪迹。

np.asarray([[[1,2,3],
             [4,5,6],
             [7,8,9]],
           [[10,11,12],
            [13,14,15],
            [16,17,18]]])

结果应为：

np.asarray([15, 42])

Answer 1

这是沿第二个轴使用advanced indexing和sum的一种方式：

a = np.asarray([[[1,2,3],
                 [4,5,6],
                 [7,8,9]],
               [[10,11,12],
                [13,14,15],
                [16,17,18]]])

a[:, range(a.shape[1]), range(a.shape[2])].sum(1)
# array([15, 42])

Answer 2

在axis1=1中使用axis2=2和np.trace自变量。

np.trace(x, axis1=1, axis2=2)

Answer 3

将np.trace函数与参数axis1=1和axis2=2一起使用

mat = np.asarray([[[1,2,3],
             [4,5,6],
             [7,8,9]],
           [[10,11,12],
            [13,14,15],
            [16,17,18]]])

np.trace(mat, axis1=1, axis2=2)

输出：

array([15, 42])