NUMPY学习系列——花哨的索引

若

a = array([0,1,2,3,4,5,6,7,8,9]),
b = array([[0,1,2,3],
           [4,5,6,7],
           [8,9,10,11]])

那么：
索引对象为向量

方法一：[x[3],x[7],x[2]]    即array([3,7,2])
方法二：ind = [3,7,2]
       x[ind]
       array([3,7,2])

索引对象为矩阵
由

row = np.array([0,1,2])
col = np.array([2,1,3])

那么：

b[row[:,np.newaxis],col]
即array([[2,1,3],
         [6,5,7],
         [10,9,11]])

注意：结果的形状与索引数组的形状一致。
组合索引
1.花哨的索引与简单的索引组合使用

b[2,[2,0,1]]

2.花哨的索引与切片的组合使用

b[1:,[2,0,1]]

3.花哨的索引与掩码的组合使用

c = np.array([1,0,1,0],dtype = bool)
b[row[:,np.newaxis],c]

返回结果是：

array([[0,2],
       [4,6],
       [8,10]])