在 python 中查找子矩阵的踪迹答案

【问题标题】：To find trace of sub-matrices in python在 python 中查找子矩阵的踪迹
【发布时间】：2017-06-11 22:00:36
【问题描述】：

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

【问题讨论】：

【解决方案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 拉出每个子矩阵，然后对其应用跟踪。这可能不是做你想做的最有效的方法，但它是透明的。

【讨论】：