python绘制三维图

在Python中，我们可以使用matplotlib库中的mplot3d工具包来绘制三维图。下面是一个简单的例子，绘制了一个三维的散点图和一个三维曲面图：

首先，确保已经安装了matplotlib库。如果没有，可以通过pip进行安装：

pip install matplotlib

然后，可以使用以下代码来绘制三维图：

import numpy as np  
import matplotlib.pyplot as plt  
from mpl_toolkits.mplot3d import Axes3D  
  
# 创建一个3D坐标轴  
fig = plt.figure()  
ax = fig.add_subplot(111, projection='3d')  
  
# 创建一个网格  
x = np.arange(-5, 5, 0.25)  
y = np.arange(-5, 5, 0.25)  
x, y = np.meshgrid(x, y)  
  
# 计算对应的z值  
r = np.sqrt(x**2 + y**2)  
z = np.sin(r)  
  
# 绘制三维曲面图  
ax.plot_surface(x, y, z, rstride=1, cstride=1, cmap='viridis', edgecolor='none')  
  
# 添加标题和坐标轴标签  
ax.set_title('3D surface plot')  
ax.set_xlabel('X axis')  
ax.set_ylabel('Y axis')  
ax.set_zlabel('Z axis')  
  
# 显示图形  
plt.show()

这段代码首先创建了一个3D坐标轴，然后创建了一个网格，并计算了每个点的z值。然后，它使用plot_surface函数绘制了一个三维曲面图。最后，它添加了标题和坐标轴标签，并显示了图形。