Encontre o primeiro índice de elemento em NumPy Array

1. Use a função where() para encontrar o primeiro índice de um elemento em um array NumPy
2. Use a função nonzero() para encontrar o primeiro índice de um elemento em um array numpy
3. Utilize a função argmax() Function to Find the First Index of an Element in a NumPy Array
4. Use a função index() para encontrar o primeiro índice de um elemento de um NumPy

Neste tutorial, discutiremos como encontrar o primeiro índice de um elemento em um array numpy.

Use a função where() para encontrar o primeiro índice de um elemento em um array NumPy

A função where() do módulo numpy é usada para retornar um array que contém os índices dos elementos que satisfazem algumas condições. A condição é especificada na função.

Podemos usá-lo para encontrar o primeiro índice de um valor específico em um array, conforme mostrado abaixo.

a = np.array([7, 8, 9, 5, 2, 1, 5, 6, 1])

print(np.where(a == 1)[0][0])

Resultado:

5

Use a função nonzero() para encontrar o primeiro índice de um elemento em um array numpy

A função nonzero() retorna os índices de todos os elementos diferentes de zero em um array numpy. Ele retorna tuplas de vários arrays para um array multidimensional.

Semelhante à função where(), podemos especificar a condição também para que ela também possa retornar a posição de um elemento específico.

Por exemplo,

a = np.array([7, 8, 9, 5, 2, 1, 5, 6, 1])

print(np.nonzero(a == 1)[0][0])

Resultado:

5

Para os propósitos mais básicos, as funções where() e diferente de zero() parecem semelhantes. A diferença surge com a função where() quando você deseja escolher elementos do array a se alguma condição for True e do array b quando essa condição for False.

Utilize a função argmax() Function to Find the First Index of an Element in a NumPy Array

O argmax() encontra o índice do elemento máximo em um array. Podemos especificar a condição de igualdade na função e encontrar o índice do elemento necessário também.

Por exemplo,

a = np.array([7, 8, 9, 5, 2, 1, 5, 6, 1])

print(np.argmax(a == 1))

Resultado:

5

Use a função index() para encontrar o primeiro índice de um elemento de um NumPy

Neste método, primeiro converteremos o array em uma lista usando a função tolist(). Em seguida, usaremos a função index(), que retornará a posição do elemento especificado.

O código a seguir implementa isso.

a = np.array([7, 8, 9, 5, 2, 1, 5, 6, 1])

print(a.tolist().index(1))

Resultado:

5