python - 如何从 numpy 二维数组中提取子数组？

我想从一个 numpy 二维数组中提取一个指定大小的 numpy 数组——本质上我想裁剪这个数组。例如，如果有一个像这样的 numpy 数组:

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

我想从中提取一个 2x2，结果应该是:

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

我该怎么做？

最佳答案

给定这个数组:

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

您可以沿两个维度对其进行切片:

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

关于python - 如何从 numpy 二维数组中提取子数组？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/35681054/