Obtenir la colonne du tableau NumPy

Ce didacticiel présentera la méthode pour obtenir une colonne spécifique à partir d’un tableau NumPy multidimensionnel en Python.

Obtenir une colonne à partir d’un tableau NumPy multidimensionnel avec la méthode de découpage de base

La méthode de découpage de base fonctionne sur le même principe que le découpage de liste en Python. Nous pouvons utiliser la méthode de découpage de base pour obtenir une colonne spécifique à partir d’un tableau NumPy multidimensionnel. La méthode de découpage de base crée une nouvelle vue de notre tableau existant au lieu de créer une nouvelle copie du tableau. Ces nouveaux points de vue sur la baie existante et la mémoire de la baie d’origine ne peuvent pas être libérés tant que toutes les vues pointant vers la baie d’origine ne sont pas collectées par le garbage collector. L’exemple de code suivant nous montre comment obtenir une colonne spécifique à partir d’un tableau NumPy multidimensionnel avec la méthode de découpage de base en Python.

import numpy as np

array = np.array([[1, 2], [3, 4], [5, 6], [7, 8], [9, 0]])
print(array[:, 1])

Production:

[2 4 6 8 0]

Dans le code ci-dessus, nous avons extrait la deuxième colonne du tableau NumPy multidimensionnel array avec l’index de découpage [:,1] en Python. La première partie de l’index est l’index des lignes. Nous avons laissé la première partie vide car nous voulons sélectionner toutes les lignes. L’opérateur : représente une opération de sélection dans l’index. Nous avons utilisé le [:,1] comme index de colonne pour obtenir la deuxième colonne de chaque ligne.