制作 3D 等高线图

Question

几天来，我一直在尝试使用 Python 获取“示例”图片（使用 3D 图形计算生成），但在显示除中间峰值以外的绘图片段时一直遇到麻烦达到规模。

我正在使用此代码：

import numpy as np
from mpl_toolkits.mplot3d import axes3d
import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111, projection="3d")
X, Y = np.mgrid[-1:1:30j, -1:1:30j]
Z = (2*X*Y) + (1/np.sqrt(X**2+Y**2))
ax.plot_surface(X, Y, Z, cmap="autumn_r", lw=0.5, rstride=1, cstride=1)
ax.contour(X, Y, Z, 10, lw=3, cmap="autumn_r", linestyles="solid", offset=-1)
ax.contour(X, Y, Z, 10, lw=3, colors="k", linestyles="solid")
plt.show()

生成此图。

它很接近，但它应该看起来更像这个。当我降低 30j 以试图降低它并希望两侧的耀斑更加明显时，它会摆脱整个峰值。我正在努力解决这个问题。

Answer 1

如果您尝试以下行怎么办？

X, Y = np.mgrid[-7:7:100j, -7:7:100j]