python matplotlib 实现3d模型

Python的Matplotlib库提供了一个mpl_toolkits.mplot3d模块，可以用于绘制3D图形。在这个模块中，有一个Axes3D类，可以用于创建3D图形的坐标系。通过定义x、y、z轴的数据，可以使用plot_surface()函数绘制3D图形。同时，可以使用cm模块中的颜色映射函数来设置3D图形的颜色。下面是一个简单的例子：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm

# 创建画布和子图
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 定义x、y、z轴的数据
x = np.linspace(-1, 1, 100)
y = np.linspace(-1, 1, 100)
x, y = np.meshgrid(x, y)
z = np.sin(np.sqrt(x**2 + y**2))

# 定义颜色
colors = np.sin(np.arctan2(y, x))

# 绘制3D图形
surf = ax.plot_surface(x, y, z, facecolors=cm.jet(colors), linewidth=0, antialiased=False)

# 调整图形视角和显示范围
ax.view_init(elev=30, azim=45)
ax.set_xlim([-1, 1])
ax.set_ylim([-1, 1])
ax.set_zlim([-1, 1])

# 显示图形
plt.show()

这个例子中，我们创建了一个100x100的网格，然后计算了每个点的z值，最后使用plot_surface()函数绘制了一个3D图形。同时，我们使用了颜色映射函数来设置3D图形的颜色。