Matplotlib - 3D 曲面图

曲面图显示指定因变量 (Y) 和两个自变量(X 和 Z)之间的函数关系。 该图是等值线图的配套图。 曲面图类似于线框图,但线框的每个面都是一个填充的多边形。 这有助于感知被可视化的表面的拓扑结构。 plot_surface() 函数 x、y 和 z 作为参数。

from mpl_toolkits import mplot3d
import numpy as np
import matplotlib.pyplot as plt
x = np.outer(np.linspace(-2, 2, 30), np.ones(30))
y = x.copy().T # transpose
z = np.cos(x ** 2 + y ** 2)

fig = plt.figure()
ax = plt.axes(projection='3d')

ax.plot_surface(x, y, z,cmap='viridis', edgecolor='none')
ax.set_title('Surface plot')
plt.show()

上面这行代码会产生如下输出 −

曲面图