如何使用 matplotlib 创建圆形 2D 图，其中函数取决于与图像中心的距离？

Question

我有一个一维的函数（如下图所示）。 我正在检查所有 matplotlib 教程，但我找不到在二维图中绘制一维数据的解决方案，其中与图像中心距离相同的所有点都具有相同的值。

从这样的功能：

我想得到这样的东西：

我正在尝试使用 Axes3D 和 imshow，但它们需要 (x,y) 坐标而不是 r（距中心的距离）。 谢谢！

Answer 1

本质上，您需要在笛卡尔网格上评估您的函数。 因此，不是使用半径r func(r)调用您的函数，而是使用从网格节点x和y计算的半径来调用它， func(sqrt(x**2+y**2))

import numpy as np
import matplotlib.pyplot as plt

func = lambda r: np.sin(r)*np.exp(-r/10.)
r = np.linspace(0,50,151)

X,Y = np.meshgrid(np.linspace(-50,50,301),np.linspace(-50,50,301))
rad = lambda x,y: np.sqrt(x**2+y**2)
image = func(rad(X,Y)) 

fig, (ax,ax2) = plt.subplots(ncols=2)

ax.plot(r, func(r))

ax2.imshow(image, extent=[X.min(),X.max(),Y.min(),Y.max()])
plt.show()