如何從numpy 2d數組中提取子數組？

Question

我想從numpy二維數組中提取具有指定大小的numpy數組 - 本質上我想裁剪該數組。 例如，如果有一個numpy的數組是這樣的：

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

我想從中提取一個2x2和結果應該是：

([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]])