小编cou*_*ode的帖子

如何从给定指示符列向量的二维数组中检索列向量？

假设我有

X = np.array([[1, 4, 6],
              [8, 2, 9],
              [0, 3, 7],
              [6, 5, 1]])

和

S = np.array([0, 2, 1, 2])

有没有一种优雅的方式来获得X结果S，array([1, 9, 3, 1])这相当于

np.array([x[s] for x, s in zip(X, S)])