Comment extraire des colonnes spécifiques de tableaux multidimensionnels dans NumPy ?

Récupération de colonnes à partir de tableaux multidimensionnels dans NumPy

Dans NumPy, une opération fondamentale consiste à accéder à des éléments individuels ou à des sous-ensembles de données dans un tableau multidimensionnel. Bien que la récupération de lignes à l'aide du test de syntaxe[i] soit simple, l'extraction de colonnes spécifiques peut être légèrement différente.

Pour accéder à la ième colonne d'un tableau NumPy, vous pouvez utiliser le test de syntaxe[:, i]. Cela renverra un tableau unidimensionnel contenant les valeurs de la ième colonne de la matrice d'origine.

Par exemple, considérons le tableau suivant :

test = np.array([[1, 2], [3, 4], [5, 6]])

Pour obtenir la première colonne (contenant les valeurs 1, 3 et 5), vous utiliseriez :

>>> test[:, 0]
array([1, 3, 5])

De même, pour récupérer la deuxième colonne (contenant les valeurs 2, 4 et 6), vous utiliseriez :

>>> test[:, 1]
array([2, 4, 6])

Notez que cette opération n'est pas particulièrement coûteuse en termes de calcul et constitue une technique courante pour travailler avec des données multidimensionnelles dans NumPy.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!