Jak uzyskać określoną kolumnę z tablicy numpy (z przykładami)

Aby pobrać konkretną kolumnę z tablicy NumPy, możesz użyć następującej składni:

 #get column 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 kolumnę z tablicy NumPy

Poniższy kod pokazuje, jak uzyskać określoną kolumnę 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 column in index position 2
data[:, 2]

array([3,7,11])

Jeśli chcesz pobrać kolumnę z tablicy NumPy i pobrać ją jako wektor kolumnowy , możesz użyć następującej składni:

 #get column in index position 2 (as a column vector)
data[:, [2]]

array([[ 3],
       [7],
       [11]])

Przykład 2: Pobierz wiele kolumn z tablicy NumPy

Poniższy kod pokazuje, jak uzyskać wiele kolumn 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 columns in index positions 1 and 3 from NumPy array
data[:, [1,3]]

array([[ 2, 4],
       [6, 8],
       [10, 12]])

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

Poniższy kod pokazuje, jak pobrać kolumny 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 columns in index positions 0 through 3 (not including 3)
data[:, 0:3]

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

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

Dodatkowe zasoby

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

Jak zmapować funkcję na tablicę NumPy
Jak dodać kolumnę do tablicy NumPy