从 Python 中非常大的 numpy ndarray 中获取 3*3 子集矩阵

Question

我试图从一个非常大的 400 x 500 ndarray 的 numpy 中获取一个 3*3 的子集。但由于某种原因，我没有得到想要的结果。而是将前三行作为一个整体。

这是我写的代码。

subset_matrix = mat[0:3][0:3]

但这就是我在 Jupyter Notebook 的输出中得到的结果

array([[91, 88, 87, ..., 66, 75, 82],
       [91, 89, 88, ..., 68, 78, 84],
       [91, 89, 89, ..., 72, 80, 87]], dtype=uint8)

Answer 1

mat[0:3][0:3] 将二维数组的轴 0 切片两次，相当于mat[0:3]。你需要的是mat[0:3,0:3]。