python - 二维函数的俯视图

Question

我正在尝试可视化与此类似的 2d 函数：.

我发现最好的方法是使用matplotlib.pyplot.contourf，但是对于非常大的函数，这非常慢，而且之前必须设置级别数。

X, Y = np.meshgrid(x, y)  # grid of point
Z = fun(X, Y)

levels = np.arange(0, 25)
im = axes.contourf(X, Y, Z, levels, zdir='z', cmap=cm.viridis)
plt.colorbar(im)

有没有“更好”更有效的方法来做到这一点？

图片来自维基百科：

.

标签： pythonmatplotlib