如何用Python分析STEP/STL格式CAD文件并提取尺寸？求最优方案

针对批量处理STEP/STL格式的平板带孔模型，提取长宽高及孔洞半径的需求，以下是具体方案分析：

numpy-stl vs pyvista 对比 #

numpy-stl #

• 适用场景：仅处理STL格式、追求轻量低依赖的场景
• 优势：基于numpy，读取STL速度快，内存占用低
• 劣势：无内置几何分析工具，所有尺寸提取、孔洞识别逻辑需手动实现（如遍历顶点计算 bounding box、分析网格拓扑识别孔洞），开发成本高，且不支持STEP格式

pyvista #

• 适用场景：同时处理STEP/STL、需要快速实现几何分析的场景
• 优势：基于VTK，支持多格式导入；内置bounds属性直接获取模型 bounding box（对应平板长宽高）；提供特征边提取、形状拟合工具，可快速识别孔洞并计算半径
• 劣势：依赖VTK，安装包体积较大，入门需熟悉VTK基本概念，但内置工具能大幅减少代码量

结论：若需同时处理STEP和STL，pyvista是更优选择；仅处理STL且资源受限，可考虑numpy-stl，但需自行开发分析逻辑。

其他可行方案 #

1. PythonOCC（基于OpenCASCADE） #

工业级CAD处理方案，专门针对STEP这类边界表示（B-Rep）格式，可直接访问模型的参数化设计特征（如平板的草图尺寸、孔洞的原始半径），精度远高于网格拟合，适合对参数精度要求高的场景。

2. FreeCAD Python脚本 #

借助开源CAD软件FreeCAD的Python API，可批量导入STEP/STL模型，通过Shape对象直接获取模型几何参数，适合已有FreeCAD使用经验的用户。

3. MeshLab 批量脚本 #

针对纯STL网格文件，可通过MeshLab的MLX脚本实现批量处理，自动计算 bounding box、拟合孔洞半径，但不支持STEP格式。

快速实现示例（pyvista） #

import pyvista as pv

# 读取STL/STEP文件
mesh = pv.read("your_model.stl")  # 替换为step文件路径也可

# 提取平板长宽高
bounds = mesh.bounds
length = bounds[1] - bounds[0]
width = bounds[3] - bounds[2]
height = bounds[5] - bounds[4]
print(f"平板尺寸：长={length:.2f}, 宽={width:.2f}, 高={height:.2f}")

# 识别并计算孔洞半径
# 提取边界边（对应孔洞边缘）
edge_mesh = mesh.extract_feature_edges(boundary_edges=True, non_manifold_edges=False)
# 拆分每个孔洞的边界环
for hole_loop in edge_mesh.split_bodies():
    # 拟合圆形并获取半径
    _, _, radius = hole_loop.fit_circle()
    print(f"孔洞半径：{radius:.2f}")

内容的提问来源于stack exchange，提问作者Rasit Er