要找到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翻出每個子矩陣，然後跟蹤適用於它。這可能不是實現你想要的最有效的方式，但它是透明的。