使用Numpy从矩阵引出所有平方子矩阵

Question

说我有一个NxN numpy矩阵。 我正在寻找从此矩阵中提取所有正方形块（子矩阵）的最快方法。 表示原始矩阵的所有CxC部分都为0 <C <N + 1 。 子矩阵应对应于原始矩阵的连续行/列索引。 我想在尽可能短的时间内实现这一目标。

Answer 1

您可以使用Numpy切片，

import numpy as np

n = 20
x = np.random.rand(n, n)

slice_list = [slice(k, l) for k in range(0, n) for l in range(k, n)]

results = [x[sl,sl] for sl in slice_list]

避免Numpy中的循环并不是一个目标。 只要您对此有所注意，就不会有太多开销。

Answer 2

足够棘手，但这是在NxN矩阵中提取所有MxM子矩阵的示例。

import numpy as NP
import numpy.random as RNG

P = N - M + 1
x = NP.arange(P).repeat(M)
y = NP.tile(NP.arange(M), P) + x

a = RNG.randn(N, N)
b = a[NP.newaxis].repeat(P, axis=0)
c = b[x, y]
d = c.reshape(P, M, N)
e = d[:, NP.newaxis].repeat(P, axis=1)
f = e[:, x, :, y]
g = f.reshape(P, M, P, M)
h = g.transpose(2, 0, 3, 1)

for i in range(0, P):
    for j in range(0, P):
        assert NP.equal(h[i, j], a[i:i+M, j:j+M]).all()