我怎样才能索引这个numpy数组？

Question

给定形状a的numpy矩阵(5,3)和形状b的索引向量(5,)，其中索引向量中的每个条目都在0到2之间，如何根据c及其索引向量a创建新的向量b。

Answer 1

将arange用于其他维度：

c = a[np.arange(5), b]