Come ottenere una colonna specifica da un array numpy (con esempi)

È possibile utilizzare la seguente sintassi per ottenere una colonna specifica da un array NumPy:

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

Gli esempi seguenti mostrano come utilizzare questa sintassi nella pratica.

Esempio 1: ottieni una colonna dall’array NumPy

Il codice seguente mostra come ottenere una colonna specifica da un 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 column in index position 2
data[:, 2]

array([3,7,11])

Se desideri ottenere una colonna da un array NumPy e recuperarla come vettore di colonna , puoi utilizzare la seguente sintassi:

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

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

Esempio 2: ottieni più colonne dall’array NumPy

Il codice seguente mostra come ottenere più colonne da un 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 columns in index positions 1 and 3 from NumPy array
data[:, [1,3]]

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

Esempio 3: ottieni colonne nell’intervallo dall’array NumPy

Il codice seguente mostra come ottenere colonne in un intervallo da un 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 columns in index positions 0 through 3 (not including 3)
data[:, 0:3]

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

Tieni presente che l’ultimo valore nell’intervallo (in questo caso 3) non è incluso nell’intervallo di colonne restituito.

Risorse addizionali

I seguenti tutorial spiegano come eseguire altre operazioni comuni in NumPy:

Come mappare una funzione su un array NumPy
Come aggiungere una colonna a un array NumPy