Utilisation du tableau Numpy comme indices pour la 2ème dimension d'un autre tableau

pour extraire des éléments spécifiques à partir d'un tableau 2D basé sur des indices fournis par un deuxième array, vous pouvez leader de l'indexage en entrée de numpy. Exemple:

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()]

• Le tableau d'index combiné [np.arange (a.shape [0]), b.ravel ()] spécifie la ligne et les indices de colonne dans un pour extraire.
.Ravel ():
a [np.arange (a.shape [0]), b]

Cette méthode fournit une approche simple pour extraire des éléments d'un tableau 2D à l'aide d'indices dérivés d'un autre tableau.