このnumpy配列のインデックスを作成するにはどうすればよいですか？

Question

形状(5,3)のnumpyの行列a、インデックスベクトルの各エントリは0〜2の間である形状(5,)のインデックスベクトルb、考えると、どのようにaとそのインデックスベクトルbに基づいて、新しいベクトルcを作成することができます。

Answer 1

他の次元に使用arange：

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