您的位置:宽带测速网 > 编程知识 > Matplotlib中怎么绘制三维散点图和曲面图

Matplotlib中怎么绘制三维散点图和曲面图

2025-06-27 17:07来源:互联网 [ ]

要绘制三维散点图和曲面图,可以使用Matplotlib中的mplot3d模块。下面分别介绍如何绘制三维散点图和曲面图:

    绘制三维散点图:
import matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3Dimport numpy as np# 生成随机数据x = np.random.rand(100)y = np.random.rand(100)z = np.random.rand(100)# 创建一个三维坐标轴fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制散点图ax.scatter(x, y, z)# 设置坐标轴标签ax.set_xlabel('X')ax.set_ylabel('Y')ax.set_zlabel('Z')plt.show()
    绘制曲面图:
import matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3Dimport numpy as np# 生成数据X = np.linspace(-5, 5, 100)Y = np.linspace(-5, 5, 100)X, Y = np.meshgrid(X, Y)Z = np.sin(np.sqrt(X**2 + Y**2))# 创建一个三维坐标轴fig = plt.figure()ax = fig.add_subplot(111, projection='3d')# 绘制曲面图ax.plot_surface(X, Y, Z, cmap='viridis')# 设置坐标轴标签ax.set_xlabel('X')ax.set_ylabel('Y')ax.set_zlabel('Z')plt.show()

以上代码分别展示了如何绘制三维散点图和曲面图。在绘制曲面图时,首先使用np.meshgrid()生成网格数据,然后使用ax.plot_surface()绘制曲面图。