我该如何从NumPy 2维数组中提取子数组？

Question

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

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

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

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

我该怎么做呢？

- mengmengxyz

---

你可以使用 numpy 索引 array[:2,:2]。 - Mazdak

Answer 1

考虑下面这个数组：

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

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

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

- Mike Müller