二维高斯光束图

Question

我想使用matplotlib构建二维高斯光束e^(-x^2)。

我需要获得：

Answer 1

您需要使用np.meshgrid来构造您的 二维函数：

import numpy as np
import matplotlib.pyplot as plt


x = np.linspace(-5, 5, 250)
y = np.linspace(-5, 5, 250)

xx, yy = np.meshgrid(x, y)

zz = np.exp(-(xx**2 + yy**2))

plt.imshow(zz)
plt.show()