如何从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]])