Code:
import numpy as np def f(a): return np.array([0, 1]) N_x, N_y = 4, 3 U = V = np.zeros((N_x, N_y)) for n_y in range(N_y): for n_x in range(N_