を使用して、numpyアレイを別の配列の2番目の次元のインデックスとして使用して

2番目のアレイによって提供されるインデックスに基づいて2Dアレイから特定の要素を抽出します。例：

a = np.array（[[0,1]、[2,3]、[4,5]]） B = np.Array（[[1]、[0]、[1]]、dtype = 'int'）

A = np.array([[0,1], [2,3], [4,5]])
B = np.array([[1], [0], [1]], dtype='int')

c = np.array（[[1]、[2]、[5]）

C = np.array([[1], [2], [5]])

a [np.arange（a.shape [0]）、b.ravel（）]

A[np.arange(A.shape[0]),B.ravel()]

Blattens B、追加の寸法を削除します。 .ravel（）： a [np.arange（a.shape [0]）、b]

この方法は、別のアレイから派生したインデックスを使用して2Dアレイから要素を抽出するための簡単なアプローチを提供します。