x = np.array([[1,2],[3,4]])
>>x
array([[1, 2],
[3, 4]])
y = np.array([[1],[0]])
>>y
array([[1],
[0]])
>>x[y]
array([[[3, 4]],
[[1, 2]]])
理解x[y]就是array([[x[1],
x[0]]])
x = np.mat([[1,2],[3,4]])
>>x
matrix([[1, 2],
[3, 4]])
y = np.mat([[1],[0]])
>>y
matrix([[1],
[0]])
>>x[y]
matrix([[[3, 4]],
[[1, 2]]])