Como obter uma linha específica de um array numpy (com exemplos)

Você pode usar a seguinte sintaxe para obter uma linha específica de um array NumPy:

 #get row in index position 2 from NumPy array
my_array[2, :]

Os exemplos a seguir mostram como usar essa sintaxe na prática.

Exemplo 1: Obtenha uma linha do array NumPy

O código a seguir mostra como obter uma linha específica de um array NumPy:

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
print (data)

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get row in index position 2
data[2,:]

array([ 9, 10, 11, 12])

Observe que apenas a linha na posição de índice 2 da matriz NumPy é retornada.

Exemplo 2: Obtenha várias linhas do array NumPy

O código a seguir mostra como obter várias linhas de um array NumPy:

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get rows in index positions 0 and 2 from NumPy array
data[[0,2],:]

array([[ 1, 2, 3, 4],
       [9, 10, 11, 12]])

Exemplo 3: Obtenha linhas no intervalo do array NumPy

O código a seguir mostra como obter linhas em um intervalo de um array NumPy:

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

#view NumPy array
data

array([[ 1, 2, 3, 4],
       [5, 6, 7, 8],
       [9, 10, 11, 12]])

#get rows in index positions 0 through 1
data[0:2, :]

array([[1, 2, 3, 4],
       [5, 6, 7, 8]])

Observe que o último valor do intervalo (neste caso, 2) não está incluído no intervalo de linhas retornado.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns no NumPy:

Como obter uma coluna específica de um array NumPy
Como mapear uma função para um array NumPy
Como adicionar uma coluna a um array NumPy