如何在已有对象下方创建新对象？复杂数据集优化方案需求

任务要求 #

• 输入数据格式：[(x1, y1, z1, x2, y2, z2), (...),...]，每个元组代表一条线段的起点（x1,y1,z1）和终点（x2,y2,z2），z值对应层级
• 绘制规则：按顺序创建层级对象，新对象隐藏在已有对象下方——先创建最高层级的obj_1，再创建obj_2（位于obj_1下方），最后创建obj_3（位于obj_1和obj_2下方），最终输出所有可见线条的坐标，效果需匹配目标样式
• 现有代码仅支持两个对象，无法适配复杂数据集，需优化

优化后代码 #

from shapely.geometry import LineString, Polygon, unary_union
from shapely.ops import batched
import matplotlib.pyplot as plt
from shapely.plotting import plot_line

# 加载数据集（将目标数据集内容粘贴到此处）
dataset = [
    # 示例坐标，替换为实际数据集内容
    (0.09, 0.98, 3, 0.13, 0.98, 3),
    (0.13, 0.98, 3, 0.17, 0.97, 3),
    # 其余坐标条目...
]

# 按z值拆分三个层级的对象（z=3为obj_1，z=2为obj_2，z=1为obj_3，根据实际数据集调整）
obj_1 = [[x1, y1, x2, y2] for x1, y1, z1, x2, y2, z2 in dataset if z1 == 3 and z2 == 3]
obj_2 = [[x1, y1, x2, y2] for x1, y1, z1, x2, y2, z2 in dataset if z1 == 2 and z2 == 2]
obj_3 = [[x1, y1, x2, y2] for x1, y1, z1, x2, y2, z2 in dataset if z1 == 1 and z2 == 1]

def create_geometry(coords):
    """将坐标转换为Shapely几何对象，闭合环转为Polygon，普通线段保留LineString"""
    geometries = []
    for coo in coords:
        line = LineString(batched(coo, 2))
        geometries.append(Polygon(line) if line.is_ring else line)
    return unary_union(geometries)

# 创建各层级的合并几何对象
geo1 = create_geometry(obj_1)
geo2 = create_geometry(obj_2)
geo3 = create_geometry(obj_3)

# 计算各层级的可见区域：下层对象减去上层所有对象的覆盖范围
visible_geo3 = geo3.difference(unary_union([geo1, geo2]))
visible_geo2 = geo2.difference(geo1)
visible_geo1 = geo1

# 绘图（按从下到上顺序绘制，确保层级遮挡正确）
fig, ax = plt.subplots(figsize=(4, 7))
plot_line(visible_geo3, color="w", add_points=False, ax=ax)
plot_line(visible_geo2, color="w", add_points=False, ax=ax)
plot_line(visible_geo1, color="w", lw=2, add_points=False, ax=ax)

# 设置样式
ax.set(facecolor="#242c34", xticks=[], yticks=[])

plt.show()

关键改进点 #

• 层级逻辑完善：支持三个层级对象的处理，按从下到上的顺序绘制，保证遮挡关系正确
• Z轴层级划分：通过输入数据中的z值筛选对应层级的对象，适配数据集的层级定义
• 可见区域计算：使用unary_union合并上层所有对象，再通过difference得到下层对象的可见部分，解决复杂遮挡问题
• 几何兼容性：自动识别闭合环并转换为Polygon，确保图形处理的准确性

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