How do I extract a sub-array from a numpy 2d array? [duplicate]

Answer 1

Given this array:

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

You can slice it along both dimensions:

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