A função argwhere () em Numpy é usada para encontrar os índices de elementos diferentes de zero em uma matriz, agrupada por componente (índices).
Parece confuso? Fique ligado ao longo deste tutorial para explorar mais.
Função Sintaxe
A função tem uma sintaxe incrivelmente simples, como mostrado na definição abaixo:
1 | Numpy.Argwhere (a) |
Como mostrado, a função leva apenas um parâmetro:
A função retornará os índices dos elementos diferentes de zero na matriz agrupada pelo elemento.
A matriz resultante contém a forma (n, a.ndim), onde n é o número de elementos diferentes de zero e um.NDIM é o número de dimensões da matriz da matriz de entrada.
Exemplo 1
Pegue o código de exemplo mostrado abaixo:
1 2 3 4 5 6 7 | # importar numpy importar numpy como np # GERE ARRAY: arr = np.ARANGE (10).remodelar (2,5) Imprimir (arr) # Índice de retorno de elementos diferentes de zero Imprimir (NP.argwhere (arr)) |
No código de exemplo acima, começamos importando Numpy.
Em seguida, criamos uma matriz usando a função de organização e a remodelamos na forma de (2,5).
Finalmente, obtemos os índices dos elementos diferentes de zero usando a função Argwhere.
1 2 3 4 5 6 7 8 9 | [[0 1] [0 2] [0 3] [0 4] [1 0] [1 1] [1 2] [1 3] [1 4]] |
Exemplo #2
Você também pode dizer a função de retornar os índices que correspondem a uma condição específica. Por exemplo, para obter o índice maior ou igual a 3, podemos fazer o seguinte:
1 | Imprimir (NP.argwhere (arr> = 3)) |
A saída resultante:
1 2 3 4 5 6 7 | [[0 3] [0 4] [1 0] [1 1] [1 2] [1 3] [1 4]] |
Conclusão
Este artigo fornece conhecimento fundamental do uso da função Argwhere () em Numpy. Sinta -se à vontade para explorar os documentos para mais.
Codificação feliz!!