import numpy as np arr = np.zeros((5, 5)) arr += np.arange(5) print(arr)