Encuentre el primer índice de elemento en NumPy Array

1. Utilice la función where() para encontrar el primer índice de un elemento en un array NumPy
2. Utilice la función distinto de cero() para encontrar el primer índice de un elemento en un array NumPy
3. Utilice la función argmax() para encontrar el primer índice de un elemento en un array NumPy
4. Utilice la función index() para encontrar el primer índice de un elemento en un array NumPy

En este tutorial, discutiremos cómo encontrar el primer índice de un elemento en un array numpy.

Utilice la función where() para encontrar el primer índice de un elemento en un array NumPy

La función where() del módulo numpy se usa para devolver un array que contiene los índices de elementos que satisfacen algunas condiciones. La condición se especifica dentro de la función.

Podemos usarlo para encontrar el primer índice de un valor específico en un array, como se muestra a continuación.

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

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

Producción :

5

Utilice la función distinto de cero() para encontrar el primer índice de un elemento en un array NumPy

La función nonzero() devuelve los índices de todos los elementos distintos de cero en un array numpy. Devuelve tuplas de múltiples para un array multidimensional.

Similar a la función where(), también podemos especificar la condición para que también pueda devolver la posición de un elemento específico.

Por ejemplo,

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

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

Producción :

5

Para los propósitos más básicos, las funciones where() y nonzero() parecen similares. La diferencia surge con la función where() cuando desea elegir elementos del array a si alguna condición es True y del array b cuando esa condición es False.

Utilice la función argmax() para encontrar el primer índice de un elemento en un array NumPy

El argmax() encuentra el índice del elemento máximo en un array. Podemos especificar la condición de igualdad en la función y también encontrar el índice del elemento requerido.

Por ejemplo,

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

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

Producción :

5

Utilice la función index() para encontrar el primer índice de un elemento en un array NumPy

En este método, primero convertiremos el array en una lista usando la función tolist(). Luego usaremos la función index(), que devolverá la posición del elemento especificado.

El siguiente código implementa esto.

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

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

Producción :

5