numpy的高級索引

Question

說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] 

解決你的問題