Numpy (16)

How to swap two columns in a 2d numpy array?

# Input
arr = np.arange(9).reshape(3,3)
arr

# Solution
arr[:, [1,0,2]]

# output
array([[1, 0, 2],
       [4, 3, 5],
       [7, 6, 8]])