Hoe u een specifieke kolom uit een numpy-array kunt halen (met voorbeelden)

U kunt de volgende syntaxis gebruiken om een specifieke kolom uit een NumPy-array te halen:

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

De volgende voorbeelden laten zien hoe u deze syntaxis in de praktijk kunt gebruiken.

Voorbeeld 1: Haal een kolom op uit de NumPy-array

De volgende code laat zien hoe u een specifieke kolom uit een NumPy-array kunt halen:

 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])

Als u een kolom uit een NumPy-array wilt halen en deze als kolomvector wilt ophalen, kunt u de volgende syntaxis gebruiken:

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

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

Voorbeeld 2: Haal meerdere kolommen op uit de NumPy-array

De volgende code laat zien hoe u meerdere kolommen uit een NumPy-array kunt halen:

 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]])

Voorbeeld 3: Haal kolommen binnen bereik op uit de NumPy-array

De volgende code laat zien hoe u kolommen in een bereik uit een NumPy-array kunt halen:

 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]])

Houd er rekening mee dat de laatste waarde in het bereik (in dit geval 3) niet is opgenomen in het geretourneerde kolombereik.

Aanvullende bronnen

In de volgende tutorials wordt uitgelegd hoe u andere veelvoorkomende bewerkingen in NumPy kunt uitvoeren:

Hoe een functie aan een NumPy-array toe te wijzen
Hoe u een kolom aan een NumPy-array toevoegt