How to swap columns of a given NumPy array?

Introduction

Swapping columns in a NumPy array is a common operation in data manipulation tasks. Whether you're preparing data for machine learning models or performing exploratory data analysis, understanding how to swap columns efficiently is crucial. In this guide, we'll explore various methods to swap columns in a NumPy array.

Using Advanced Indexing

NumPy's advanced indexing allows you to reorder columns by specifying their indices. This method is straightforward and does not require additional functions.

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

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

Using np.swapaxes()

The np.swapaxes() function swaps two axes of an array. By transposing the array, swapping the axes, and transposing it back, you can swap columns.

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
arr = arr.T
arr[[0, 1]] = arr[[1, 0]]
arr = arr.T
print(arr)

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

Using np.transpose()

The np.transpose() function reverses the axes of an array. By transposing the array, swapping the rows, and transposing it back, you can swap columns.

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
arr = arr.T
arr[[0, 1]] = arr[[1, 0]]
arr = arr.T
print(arr)

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

Using Temporary Variables

By using temporary variables, you can swap columns without affecting the original data.

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
temp = arr[:, 0].copy()
arr[:, 0] = arr[:, 1]
arr[:, 1] = temp
print(arr)

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

Conclusion

Swapping columns in a NumPy array can be accomplished through various methods, each suitable for different scenarios. Advanced indexing offers a concise approach, while functions like np.swapaxes() and np.transpose() provide flexibility for more complex operations. Using temporary variables ensures data integrity during the swap. Understanding these techniques allows you to manipulate NumPy arrays effectively for your data analysis tasks.