import numpy as np
train=np.array([1,2,3,4])
print(train)
u=np.array([[1,22],[1,20],[2,55],[2,45],[4,30],[5,66]])
print(u)
print(u[train])
import numpy as np
train=np.array([1,2,3,4])
print(train)
u=np.array([[1,22],[1,20],[2,55],[2,45],[4,30],[5,66]])
print(u)
print(u[train])