Numpy NP.Argwhere

Numpy NP.Argwhere

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:

  1. a - refere -se à matriz de entrada ou ao objeto ARRAY_LIE.

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!!