使用ravel()
a = np.array(np.random.randint(0,20,(15,2)))
b = np.array(np.random.randint(0,2,(15,1))).astype(np.float32)
#若果使用a[b==1]就会报错,使用ravel()转化成以为数组
a[b.ravel()==1]
Out[38]:
array([[ 6, 4],
[19, 1],
[ 4, 6],
[ 1, 4],
[17, 2],
[ 1, 4],
[ 4, 7],
[ 9, 2]])