usando a matriz numpy como índices para a 2ª dimensão de outra matriz

para extrair elementos específicos de uma matriz 2D com base nos índices fornecidos por uma segunda matriz, você pode ser atingido. Exemplo:

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]]] Método:

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

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

np.arange (a.shape [0]) Aplica B, removendo quaisquer dimensões adicionais. .ravel ():

a [np.arange (a.shape [0]), b]
Este método fornece uma abordagem direta para extrair elementos de uma matriz 2D usando índices derivados de outra matriz.