NumPy расширенных индексаций на многомерном массиве

Question

Say x является 3x3 NumPy массивом, который содержит следующее:

import numpy as np 
x = np.array([[ 1., 2., 3.], 
       [ 4., 5., 6.], 
       [ 7., 8., 9.]]) 

есть некоторые индексации, которые могут дать мне следующую подмассив:

array([[ 1., 2.], 
     [ 5., 6.]]) 

Answer 1

Вы можете использовать integer array indexing с кортежем массивов:

>>> rows = np.array([[0, 0], 
...     [1, 1]], dtype=np.intp) 
>>> columns = np.array([[0, 1], 
...      [1, 2]], dtype=np.intp) 
>>> x[rows, columns] 
array([[ 1., 2.], 
     [ 5., 6.]]) 

Answer 2

вы можете использовать

x[:2,:2] 

, чтобы решить вашу проблему