Numpyic Ways вывести первые n rows и столбцы из каждых рядов и столбцов из квадратной матрицы ⇐ Python

[Anonymous]

У меня есть 20 -х 20 квадратная матрица. Я хочу взять первые 2 ряда и столбцов из каждых 5 рядов и столбцов, что означает, что вывод должен быть 8 x 8 квадратной матрицей. Это можно сделать в 2 шагах подряд следующим образом: < /p>
import numpy as np

m = 5
n = 2
A = np.arange(400).reshape(20,-1)
B = np.asarray([row for i, row in enumerate(A) if i % m < n])
C = np.asarray([col for j, col in enumerate(B.T) if j % m < n]).T
< /code>
Однако я ищу эффективность. Есть ли более Numpyic Way для этого? Я бы предпочел сделать это за один шаг.


Подробнее здесь: https://stackoverflow.com/questions/717 ... nd-columns