Numpy (17)

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

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

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

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