如何用Matplotlib绘制STL文件并匹配MATLAB可视化效果？

我能通过以下MATLAB代码轻松绘制STL文件，得到理想的脊柱模型可视化效果：

phacon = stlread("Spine_PhaconModel.stl");

hold on

figure = trisurf(phacon,'FaceColor',[0.6 0.6 0.6], 'Edgecolor','none','FaceLighting','gouraud','AmbientStrength', 0.15, 'MarkerEdgeColor',"r")

view([180, -1])

camlight('headlight');

material('dull');

但转换为Python后效果不符合预期，我尝试用mesh.Mesh.from_file读取STL数据，plot_trisurf绘图，代码如下：

fig = plt.figure()

ax = fig.gca(projection='3d')

stl_data = mesh.Mesh.from_file('Spine_PhaconModel.stl')

points = stl_data.points.reshape([-1, 3])

x = points[:,0]

y = points[:,1]

z = points[:,2]

collec = ax.plot_trisurf(x,y,z,linewidth=0.2)

需要调整Python代码参数，达到和MATLAB一致的可视化效果。

调整后的Python代码 #

要完全匹配MATLAB的渲染效果，需要对应设置颜色、光照、视角等参数，修正后的代码如下：

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from stl import mesh

# 读取STL模型
stl_data = mesh.Mesh.from_file('Spine_PhaconModel.stl')

# 提取三角形面的顶点和索引
faces = stl_data.vectors
x = faces[:, :, 0].flatten()
y = faces[:, :, 1].flatten()
z = faces[:, :, 2].flatten()
# 生成三角形面的索引，确保plot_trisurf正确识别面结构
triangle_indices = [[3*i, 3*i+1, 3*i+2] for i in range(len(faces))]

# 创建3D绘图环境
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

# 绘制三角曲面，对齐MATLAB参数
surf = ax.plot_trisurf(x, y, z, 
                       triangles=triangle_indices,
                       facecolor=(0.6, 0.6, 0.6),  # 对应MATLAB的FaceColor
                       edgecolor='none',  # 关闭边缘线
                       linewidth=0,
                       shade=True)  # 启用Gouraud着色

# 设置视角，匹配MATLAB的view([180, -1])
ax.view_init(elev=-1, azim=180)

# 配置光照，模拟MATLAB的camlight和dull材质
surf.set_ambient((0.15, 0.15, 0.15))  # 环境光强度对应AmbientStrength
surf.set_specular((0.1, 0.1, 0.1))    # 低高光模拟dull材质
ax.set_axis_off()  # 可选：隐藏坐标轴，更贴近MATLAB效果
ax.set_aspect('equal', adjustable='box')  # 保持模型比例

plt.show()

关键参数对应说明 #

• 面结构修正：直接提取STL的三角形向量并生成索引，避免顶点重复导致的绘图混乱
• 颜色与边缘：facecolor匹配MATLAB的灰度值，edgecolor='none'关闭边缘线
• 光照配置：shade=True启用Gouraud着色，set_ambient设置环境光强度，低specular值模拟哑光材质
• 视角对齐：view_init(elev=-1, azim=180)完全对应MATLAB的视角参数

内容的提问来源于stack exchange，提问作者Maria Smith