Python中查看3D STL文件切片出现空图，寻求解决方法

问题代码回顾 #

用户尝试加载STL并生成切片，但输出空图，原代码如下：

import numpy as np
from stl import mesh
import matplotlib.pyplot as plt

# 加载STL文件
mesh_data = mesh.Mesh.from_file('MyStl.stl')

# 定义切片轴（X、Y或Z）
slicing_axis = 'Y'  # 替换为所需轴

# 定义切片数量
num_slices = 10  # 根据需要调整切片数量

# 计算切片轴上的范围
axis_range = mesh_data.points[:, 2]  # 假设为Z轴切片

# 创建均匀分布的切片位置
slice_positions = np.linspace(min(axis_range), max(axis_range), num_slices)

# 遍历切片并生成2D图像
for i, slice_position in enumerate(slice_positions):
    # 定义匹配顶点的小容差
    tolerance = 1e-6

    # 创建掩码以提取切片
    if slicing_axis == 'X':
       mask = np.isclose(mesh_data.points[:, 0], slice_position, atol=tolerance)
    elif slicing_axis == 'Y':
       mask = np.isclose(mesh_data.points[:, 1], slice_position, atol=tolerance)
    elif slicing_axis == 'Z':
       mask = np.isclose(mesh_data.points[:, 2], slice_position, atol=tolerance)

    # 提取切片的顶点
    sliced_vertices = mesh_data.points[mask]

    # 创建切片的2D图（可使用任意2D绘图库）
    plt.figure()
    plt.scatter(sliced_vertices[:, 0], sliced_vertices[:, 1], s=1)
    plt.axis('equal')  # 确保绘图的宽高比一致
    plt.title(f'Slice {i+1}')
    plt.xlabel('X')
    plt.ylabel('Y')
    plt.show()

问题根源 #

1. 轴范围与切片轴不匹配：计算axis_range时硬编码用了Z轴顶点坐标（[:,2]），若选择Y/X作为切片轴，切片位置会落在错误的轴范围内，无法匹配到顶点。
2. 切片逻辑局限性：仅检查顶点是否刚好落在切片位置上，而实际切片大多与三角形边相交而非经过顶点，因此几乎提取不到任何点。
3. 绘图坐标固定错误：无论选择哪个切片轴，都固定绘制X-Y坐标，比如切片轴为Y时，应绘制X-Z坐标才能正确展示切片。

修复后的代码 #

import numpy as np
from stl import mesh
import matplotlib.pyplot as plt

# 加载STL文件
mesh_data = mesh.Mesh.from_file('MyStl.stl')

# 定义切片轴（X、Y或Z）
slicing_axis = 'Y'  # 替换为所需轴
num_slices = 10

# 根据切片轴获取对应的索引和剩余两个绘图轴的索引
axis_map = {'X':0, 'Y':1, 'Z':2}
slice_axis_idx = axis_map[slicing_axis]
plot_axes = [idx for idx in [0,1,2] if idx != slice_axis_idx]
x_plot_idx, y_plot_idx = plot_axes

# 计算当前切片轴的范围
axis_min = np.min(mesh_data.points[:, slice_axis_idx])
axis_max = np.max(mesh_data.points[:, slice_axis_idx])
slice_positions = np.linspace(axis_min, axis_max, num_slices)

def get_slice_intersections(triangle, slice_pos, axis_idx):
    """计算单个三角形与切片平面的交点"""
    v_vals = triangle[:, axis_idx]
    below = v_vals <= slice_pos
    above = v_vals >= slice_pos
    
    # 三角形完全在切片平面一侧，无交点
    if np.all(below) or np.all(above):
        return None
    
    intersections = []
    edges = [(0,1), (1,2), (2,0)]
    for v1_idx, v2_idx in edges:
        v1 = triangle[v1_idx]
        v2 = triangle[v2_idx]
        # 边跨越多切片平面时计算交点
        if (below[v1_idx] and above[v2_idx]) or (above[v1_idx] and below[v2_idx]):
            t = (slice_pos - v1[axis_idx]) / (v2[axis_idx] - v1[axis_idx])
            intersection = v1 + t * (v2 - v1)
            intersections.append(intersection)
    return np.array(intersections) if len(intersections) >=2 else None

# 遍历每个切片位置生成图像
for i, slice_pos in enumerate(slice_positions):
    slice_points = []
    # 遍历STL中的每个三角形面
    for triangle in mesh_data.points.reshape(-1,3,3):
        intersections = get_slice_intersections(triangle, slice_pos, slice_axis_idx)
        if intersections is not None:
            slice_points.extend(intersections)
    
    if not slice_points:
        print(f"Slice {i+1} 无交点")
        continue
    
    slice_points = np.array(slice_points)
    # 绘制切片的2D图
    plt.figure(figsize=(6,6))
    plt.scatter(slice_points[:, x_plot_idx], slice_points[:, y_plot_idx], s=2)
    plt.axis('equal')
    plt.title(f'Slice {i+1} (Axis {slicing_axis} = {slice_pos:.2f})')
    plt.xlabel(['X','Y','Z'][x_plot_idx])
    plt.ylabel(['X','Y','Z'][y_plot_idx])
    plt.show()

关键修改说明 #

• 轴索引动态匹配：根据选择的切片轴自动获取对应坐标索引，确保切片位置范围准确。
• 三角形边交点计算：通过判断三角形边与切片平面的交叉情况，用线性插值计算交点，覆盖所有可能的切片场景，不再依赖顶点重合。
• 动态绘图坐标：根据切片轴自动选择要绘制的两个轴，保证切片展示符合空间逻辑。

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