欢迎访问宙启技术站
智能推送

Python中使用Axes3D()绘制3D散点图

发布时间:2023-12-26 03:02:56

在Python中,可以使用matplotlib库中的Axes3D()函数来绘制3D散点图。下面是一个使用例子,将包含随机数据的三个变量绘制成三维散点图:

import numpy as np
import matplotlib.pyplot as plt

# 创建数据
num_points = 100
x = np.random.random(num_points)
y = np.random.random(num_points)
z = np.random.random(num_points)

# 创建3D图形对象
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制散点图
ax.scatter(x, y, z, c='r', marker='o')

# 设置坐标轴标签
ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('Z')

# 显示图形
plt.show()

在这个例子中,首先我们使用numpy库中的random函数生成了100个随机数作为x, y, z三个变量的值。然后通过matplotlib.pyplot库中的scatter函数绘制了这些散点。指定参数c为'r'将散点的颜色设置为红色,marker参数为'o'将散点的形状设置为圆形。

接下来,我们创建了一个3D图形对象fig,并通过add_subplot函数添加了一个三维坐标轴ax。这样就创建了一个空的三维图形对象。

然后,我们使用ax.scatter函数绘制了散点图,并对图形进行了一些美化。通过ax.set_xlabel、ax.set_ylabel和ax.set_zlabel函数设置了坐标轴的标签为'X'、'Y'和'Z'。

最后,通过plt.show函数展示了图形。

通过这个例子,我们可以看到使用Axes3D()函数可以很方便地在Python中绘制出三维的散点图。可以通过更改数据和设置函数中的参数来适应不同的需求。