Jak uzyskać określony wiersz z tablicy numpy (z przykładami)

Aby uzyskać określony wiersz z tablicy NumPy, możesz użyć następującej składni:

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

Poniższe przykłady pokazują, jak używać tej składni w praktyce.

Przykład 1: Pobierz wiersz z tablicy NumPy

Poniższy kod pokazuje, jak uzyskać określony wiersz z tablicy 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])

Należy zauważyć, że zwracany jest tylko wiersz znajdujący się na pozycji indeksu 2 tablicy NumPy.

Przykład 2: Pobierz wiele wierszy z tablicy NumPy

Poniższy kod pokazuje, jak uzyskać wiele wierszy z tablicy 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]])

Przykład 3: Pobierz wiersze z zakresu z tablicy NumPy

Poniższy kod pokazuje, jak pobrać wiersze z zakresu z tablicy 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]])

Należy pamiętać, że ostatnia wartość w zakresie (w tym przypadku 2) nie jest uwzględniana w zwracanym zakresie wierszy.

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe operacje w NumPy:

Jak uzyskać określoną kolumnę z tablicy NumPy
Jak zmapować funkcję na tablicę NumPy
Jak dodać kolumnę do tablicy NumPy