16
Como obtenho índices de N valores máximos em uma matriz NumPy?
NumPy propõe uma maneira de obter o índice do valor máximo de uma matriz via np.argmax. Gostaria de algo semelhante, mas retornando os índices dos Nvalores máximos. Por exemplo, se tiver uma matriz, [1, 3, 2, 4, 5], function(array, n=3)iria retornar os índices [4, 3, 1], que correspondem aos elementos …
484
python
numpy
max
numpy-ndarray