Я реализую алгоритм, который требует от меня взглянуть на непересекающиеся последовательные подматрицы внутри (строго двумерного) массива numpy. например, для 12 на 12
>>> a = np.random.randint(20, size=(12, 12)); a
array([[ 4, 0, 12, 14, 3, 8, 14, 12, 11, 18, 6, 6],
[15, 13, 2, 18, 15, 15, 16, 2, 9, 16, 6, 4],
[18, 18, 3, 8, 1, 15, 14, 13, 13, 13, 7, 0],
[ 1, 9, 3, 6, 0, 4, 3, 15, 0, 9, 11, 12],
[ 5, 15, 5, 6, 4, 4, 18, 13, 10, 17, 11, 8],
[13, 17, 8, 15, 17, 12, 7, 1, 13, 15, 0, 18],
[ 2, 1, 11, 12, 3, 16, 11, 9, 10, 15, 4, 16],
[19, 11, 10, 7, 10, 19, 7, 13, 11, 9, 17, 8],
[14, 14, 17, 0, 0, 0, 11, 1, 10, 14, 2, 7],
[ 6, 15, 6, 7, 15, 19, 2, 4, 6, 16, 0, 3],
[ 5, 10, 7, 5, 0, 8, 5, 8, 9, 14, 4, 3],
[17, 2, 0, 3, 15, 10, 14, 1, 0, 7, 16, 2]])
и глядя на подматрицы 3x3, я бы хотел, чтобы первая подматрица 3x3 находилась в верхнем левом углу:
>>> a[0:3, 0:3]
array([[ 4, 0, 12],
[15, 13, 2],
[18, 18, 3]])
Следующий, чтобы дать a[0:3, 3:6]
и так далее. Не имеет значения, имеет ли последний такой набор индексов в каждой строке или столбце конец конца массива - поведение numpy просто дает часть внутри существующего среза.
Я хочу, чтобы сгенерировать эти индексы среза программно для матриц произвольного размера и подматриц. В настоящее время у меня есть это:
size = 3
x_max = a.shape[0]
xcoords = range(0, x_max, size)
xcoords = zip(xcoords, xcoords[1:])
и аналогично генерировать y_coords
, так что ряд индексов задается выражением itertools.product(xcoords, ycoords)
.
Мой вопрос: есть ли более прямой способ сделать это, возможно, используя numpy.mgrid
или какой-нибудь другой метод numpy?