Numpy数组花式索引

整数数组索引

一维数组索引

import numpy as np
fi1=np.array(['Tom猫','加菲猫','波斯猫','黑猫','英国短脸猫','田园猫'])
f1=np.array([1,2,4,5])
print(fi1[f1])                     #去掉不是猫品种分类的猫

['加菲猫', '波斯猫', '英国短脸猫', '田园猫']

二维数组索引

import numpy as np
fi2=np.array([['Tom猫',1,200],['加菲猫',10,1000],['波斯猫',5,2000],['黑猫',2,180],['英国短脸猫',8,1800],['田园猫',20,100]])
f2=np.array([1,2,3])     #用一维数组指定第2，3，4行
fi2[f2]
fi3=np.array([[0,-1,9],[8,1,10],[-2,8,3]])
print(fi3)

[[ 0 -1  9]
 [ 8  1 10]
 [-2  8  3]]

x=np.array([[0,1,2]])                  #以一维数组形式，指定所有x坐标值 
y=np.array([0,1,2])                   #以一维数组形式，指定所有y坐标值
print(fi3[x,y])                            #求x,y坐标对应的所有元素

[[0 1 3]]

布尔数组索引

s4=np.arange(9).reshape(3,3)
s4
b1=np.array([[True,False,False],[False,True,False],[False,False,True]])  #布尔数组
b1
print(s4[b1])                             #布尔数组做索引

[0, 4, 8]