使用Python按照索引查找元素

#Search elements in every row using index

X = np.random.randint(1,10, [5,6])

print(X)
Y = np.array([0,3,2,4,5])

length = len(X)

B = [] 

for i in range(length):
    
    Index = np.take(X[i],Y[i])
    B.append(Index)
    print(Index)

print(B)

B = np.reshape(B,(-1, 1))

A = X / B

print(A)