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

avatar
作者
筋斗云
阅读量:3

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

  1. 绘制三维散点图:
import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D import 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() 
  1. 绘制曲面图:
import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D import 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()绘制曲面图。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!