Comment échanger deux colonnes dans un tableau NumPy (avec exemple)

Vous pouvez utiliser la syntaxe de base suivante pour échanger deux colonnes dans un tableau NumPy :

some_array[:, [0, 2]] = some_array[:, [2, 0]]

Cet exemple particulier échangera les première et troisième colonnes du tableau NumPy appelé some_array .

Toutes les autres colonnes resteront dans leur position d’origine.

L’exemple suivant montre comment utiliser cette syntaxe dans la pratique.

Connexe : Comment échanger deux lignes dans un tableau NumPy

Exemple : échanger deux colonnes dans un tableau NumPy

Supposons que nous ayons le tableau NumPy suivant :

import numpy as np

#create NumPy array
some_array = np.array([[1, 1, 2], [3, 3, 7], [4, 3, 1], [9, 9, 5], [6, 7, 7]])

#view NumPy array
print(some_array)

[[1 1 2]
 [3 3 7]
 [4 3 1]
 [9 9 5]
 [6 7 7]]

Nous pouvons utiliser la syntaxe suivante pour échanger les première et troisième colonnes du tableau NumPy :

#swap columns 1 and 3
some_array[:, [0, 2]] = some_array[:, [2, 0]]

#view updated NumPy array
print(some_array)

[[2 1 1]
 [7 3 3]
 [1 3 4]
 [5 9 9]
 [7 7 6]]

Notez que les première et troisième colonnes ont été permutées.

Toutes les autres colonnes sont restées dans leur position d’origine.

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres tâches courantes dans NumPy :

Comment supprimer les éléments en double dans le tableau NumPy
Comment remplacer des éléments dans un tableau NumPy
Comment classer les éléments dans le tableau NumPy