import numpy as np

# Create a 5x5 array with values from 0 to 24
A = np.arange(25).reshape(5,5)
print(A)

A[[0,1]] = A[[1,0]]
print("\nAfter swap:\n",A)