numpy配列インデックスを使用したエクスプレスリストの理解

Question

私は、形状（2、10）の32個のサンプルと、形状の1次元配列A（32 、）。 Aの値は0または1のいずれかで、1サンプルあたり1行目または2行目を選択するために使用されます。

result = np.array([P[i, A[i]] for i in range(32)]) 

これは単なる配列インデックスを使用して表現できますか？つまり、次のような形式の式があります。

result = P[MAGIC_HERE] 

もしそうなら、計算コストがかかりません（高速）。

Answer 1

あなたはそうのようなnumpyののインデックスを使用することができます非常に高速だった

result = P[np.arange(32), A[np.arange(32)]]