python三维图形等高线_Python/matplotlib中三维曲面与等高线的叠加

我想要一个纽比阵列绘制为一个三维表面和零轮廓线突出显示在一个引人注目的颜色。我使用以下代码绘制曲面（k_mean是要绘制的数组）：fig = plt.figure()

ax = fig.gca(projection='3d')

X,Y=np.meshgrid(range(k_mean.shape[0]), range(k_mean.shape[1]))

Z=k_mean

surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1,linewidth=0., edgecolor=None, alpha=0.8, cmap=cm.nipy_spectral, antialiased=False)

cset = ax.contour(X, Y, Z, zdir='z', offset=0, cmap=cm.nipy_spectral)

cset = ax.contour(X, Y, Z, zdir='x', offset=0, cmap=cm.nipy_spectral)

cset = ax.contour(X, Y, Z, zdir='y', offset=120, cmap=cm.nipy_spectral)

ax.set_xlim(0, 120)

ax.set_ylim(0, 120)

ax.set_zlim(0, 1)

fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()

现在，我添加了以下一行，它本身就很好：

cs=plt.contour(X,Y,Z,levels=[0], colors='g', linewidths=3)

但根据插入位置的不同，只绘制曲面图或等高线。在

如果以零轮廓线出现在曲面顶部的方式来修复它，怎么可能？在