import numpy as np
grid_x, grid_y = np.mgrid[-2:2:50j, -2:2:50j]
A = 1 # Gaussian height
x_0, y_0 = 0, 0 # Gaussian centers
sigma_x, sigma_y = 0.5, 0.5
gaussian_2D = A * np.exp(-(((grid_x - x_0)**2 / (2 * sigma_x**2)) + ((grid_y - y_0)**2 / (2 * sigma_y**2))))