当前位置：极客教程 > Matplotlib > Matplotlib 教程 > Matplotlib – 三维等高线图

Matplotlib – 三维等高线图

ax.contour3D() 函数可以创建三维等高线图。它要求所有的输入数据都是二维规则网格的形式，在每个点上都要评估Z-数据。这里，我们将展示一个三维正弦函数的三维等高线图。

from mpl_toolkits import mplot3d
import numpy as np
import matplotlib.pyplot as plt
def f(x, y):
   return np.sin(np.sqrt(x ** 2 + y ** 2))

x = np.linspace(-6, 6, 30)
y = np.linspace(-6, 6, 30)

X, Y = np.meshgrid(x, y)
Z = f(X, Y)

fig = plt.figure()
ax = plt.axes(projection='3d')
ax.contour3D(X, Y, Z, 50, cmap='binary')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('z')
ax.set_title('3D contour')
plt.show()