Dibujar mapa de contorno
Use pyplot para dibujar el mapa de contorno
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()
Resultado de salida
Dibujar en 3D
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()
Resultado de salida:
