08 Python Matplotlib 库绘制等高线和三维图

绘制等高线图

利用pyplot绘制等高线图

x = np.linspace(-10, 10, 100)
y = np.linspace(-10, 10, 100)

# 计算 x 和 y 的相交点 a
X, Y = np.meshgrid(x, y)

# 计算Z的坐标

Z = np.sqrt(X**2 + Y**2)
plt.contourf(X, Y, Z)
#或
plt.contour(X, Y, Z)

plt.show()

输出结果

绘制三维图

from mpl_toolkits.mplot3d import Axes3D

# 创建X, Y, Z 坐标
X = [1, 1, 2, 2]
Y = [3, 4, 4, 3]
Z = [1, 100, 1, 1]
fig = plt.figure()
# 创建一个Axes3D的子图放在figure画布里边
ax = Axes3D(fig)
ax.plot_trisurf(X, Y, Z)
plt.show()

输出结果：