Python教程:matplotlib中的曲面和三维等高线

Python中想要使用matplotlib使用颜色图,线框和轮廓绘制表面。 像这样:

Python教程:matplotlib中的曲面和3D轮廓

请注意,不是与xy平行的平面中的contour,而是关于图像中3D和白色的contour。

如果用幼稚的方法绘制所有这些东西,我将不到轮廓(参见下面的代码和图像)。

如果在表面添加透明度,我可以看到contour但它看起来真的很混乱(参见下面的代码和图像)

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 = np.sin(np.pi*X)*np.sin(np.pi*Y) 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()

Python教程:matplotlib中的曲面和3D轮廓

问题:有没有办法在matplotlib中获得此结果? 但是,阴影不是必需的。

如果将contour的Alpha值设置为较小的值,则看起来可能不太混乱。

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 = np.sin(np.pi*X)*np.sin(np.pi*Y) ax.plot_surface(X, Y, Z, cmap="autumn_r", lw=0.5, rstride=1, cstride=1, alpha=0.5) 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()

如果您尝试这样做,显然是一个错误

旋转一下,你会看到contour在不该消失的地方消失了

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 = np.sin(np.pi*X)*np.sin(np.pi*Y) ax.plot_surface(X, Y, Z, cmap="autumn_r", lw=0, rstride=1, cstride=1) ax.contour(X, Y, Z+1, 10, lw=3, colors="k", linestyles="solid") plt.show()

我认为您想将contour设置为偏移量:

ax.contour(X, Y, Z, 10, offset=-1, lw=3, colors="k", linestyles="solid", alpha=0.5)

Python教程:matplotlib中的曲面和3D轮廓

偏移:如果指定,则在垂直于zdir的平面上在该位置上绘制轮廓线的投影

注意,默认情况下zdir ='z',但是可以通过相应地设置zdir来在x或y方向上投影。

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/7bdb6268809c3461493f9abbb4b175e3.html