Como trocar duas colunas em um array numpy (com exemplo)

Você pode usar a seguinte sintaxe básica para trocar duas colunas em uma matriz NumPy:

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

Este exemplo específico irá trocar a primeira e a terceira colunas do array NumPy chamado some_array .

Todas as outras colunas permanecerão em suas posições originais.

O exemplo a seguir mostra como usar essa sintaxe na prática.

Relacionado: Como trocar duas linhas em um array NumPy

Exemplo: troque duas colunas em um array NumPy

Suponha que temos o seguinte array NumPy:

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

Podemos usar a seguinte sintaxe para trocar a primeira e a terceira colunas do array 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]]

Observe que a primeira e a terceira colunas foram trocadas.

Todas as outras colunas permaneceram em suas posições originais.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras tarefas comuns no NumPy:

Como remover elementos duplicados no array NumPy
Como substituir elementos em um array NumPy
Como ordenar elementos no array NumPy