Seleccione un elemento o subarray por índice de una array Numpy

Los elementos de una array NumPy se indexan como arrays normales. El índice del primer elemento será 0 y el último elemento estará indexado n-1, donde n es el número total de elementos.

Seleccionar un solo elemento de una array NumPy

Se puede acceder a cada elemento de estos ndarrays utilizando su número de índice.

Ejemplo: el siguiente código muestra cómo acceder a un elemento de una array NumPy.

import numpy as np
  
# NumPy Array
numpyArr = np.array([1, 2, 3, 4])
print("numpyArr[0] =", numpyArr[0])
print("numpyArr[-1] =", numpyArr[-1])

Producción:

numpyArr[0] = 1
numpyArr[-1] = 4

En el primer caso, accedimos al primer elemento del arreglo usando su número de índice. En el segundo caso accedimos al último elemento del arreglo usando índices negativos.

Selección de un subconjunto de un conjunto NumPy (rebanado)

Para obtener un subarreglo, pasamos un segmento en lugar del índice del elemento.

Sintaxis:

numpyArr[x:y]

Aquí x e y son el índice inicial y el último del subarreglo requerido.

Ejemplo:

import numpy as np
  
# NumPy Array
numpyArr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
  
# the slice 3:6 is passed instead 
# of index
print("Sub-Array=", numpyArr[3:6])

Producción:

Sub-Array= [4 5 6]

Se seleccionó un subarreglo a partir del 3.er índice hasta el 6.º índice (excluyendo el último, el 6.º índice). Puede dividir un subarreglo a partir del primer elemento dejando el índice inicial en blanco.

Ejemplo: el siguiente código selecciona un subarreglo a partir del primer elemento. 

import numpy as np
  
numpyArr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
  
# works same as 0:6
print("Sub-Array=", numpyArr[:6])

Producción:

Sub-Array= [1 2 3 4 5 6]

De manera similar, dejar el lado izquierdo de los dos puntos en blanco le dará una array hasta el último elemento.

Ejemplo: el siguiente código selecciona un subarreglo a partir de un índice particular hasta el último índice.

import numpy as np
  
numpyArr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
  
# same as 3:9 or 3:n, where n is
# the length of array
print("Sub-Array=", numpyArr[3:])

Producción:

Sub-Array= [4 5 6 7 8 9]