using an numpy array as indices of the 2nd dim of another array? [duplicate]

Answer 1

C = np.array([A[i][j] for i,j in enumerate(B)])