Как извлечь подматрицу из массива numpy 2d?

Question

Я хотел бы извлечь массив numpy с указанным размером из массива numpy 2d - по существу, я хочу обрезать массив. Например, если есть Numpy массив так:

([1,2,3], 
[4,5,6], 
[7,8,9]) 

я хотел бы, чтобы извлечь из нее 2х2 и результат должен быть:

([1,2], 
[4,5]) 

Как я могу это сделать?

Answer 1

Учитывая этот массив:

>>> a 
array([[1, 2, 3], 
     [4, 5, 6], 
     [7, 8, 9]]) 

Вы можете нарезать его по обоим параметрам:

>>> a[:2,:2] 
array([[1, 2], 
     [4, 5]])