在python中寻找子矩阵的踪迹

Question

我有一个更大的矩阵3Nx3N。我需要找到3x3维的每个submatrix的踪迹。我已经在SO上找到了将大矩阵拆分为更小的submatrices的答案，但我找不到一种方法来存储它们并找到Python中每个子矩阵的踪迹。任何建议都会有很大的帮助。提前谢谢你。在

Answer 1

好吧，numpy的数组上有一个trace()方法，所以您可以使用：

list( a[3*m:3*m+3,3*m:3*m+3].trace() for m in range(n) )

这里，切片3*m:3*m+3,3*m:3*m+3提取每个子矩阵，然后对其应用跟踪。这可能不是做你想做的事情的最有效的方法，但它是透明的。在