使用Python从DWG/DXF文件提取房间坐标与标签

我之前也帮人处理过类似的办公室CAD平面图提取需求，你遇到的核心问题其实是如何把房间轮廓和标签关联起来——毕竟dxfgrabber只是帮你把DXF里的元素捞出来，但不会自动帮你配对。下面给你一套实操性强的方案，一步步来：

一、先搞懂DXF里的元素关联逻辑 #

首先得明确：CAD里的房间矩形轮廓，一般是LWPOLYLINE（轻量多段线）或者POLYLINE对象，而且都是闭合的；房间标签（比如“会议室”“开放工位区”）则是TEXT或者MTEXT对象。难点在于把这两类对象绑定，常见的绑定规则有这几种：

• 标签的插入点在房间轮廓的内部（最通用，90%的CAD图都这么做）
• 标签和轮廓在同一个**图层（Layer）或者块（Block）**里
• 部分规范的CAD会用**属性块（ATTDEF/ATTRIB）**把标签和轮廓打包成一个块

二、基于dxfgrabber的具体实现步骤 #

1. 先把核心元素捞出来 #

先加载DXF，把所有闭合的轮廓和文本标签提取出来：

import dxfgrabber

# 加载目标DXF文件
dxf = dxfgrabber.readfile("your_office_plan.dxf")

# 提取所有闭合的房间轮廓（优先LWPOLYLINE，比POLYLINE更常用）
room_outlines = [
    entity for entity in dxf.entities 
    if entity.dxftype == 'LWPOLYLINE' and entity.closed
]

# 提取所有文本标签（TEXT和MTEXT都要覆盖，避免漏标签）
room_labels = []
for entity in dxf.entities:
    if entity.dxftype in ['TEXT', 'MTEXT']:
        room_labels.append({
            "label_text": entity.text,
            "insert_pos": (entity.insert[0], entity.insert[1])  # 只取XY平面坐标，忽略Z轴
        })

2. 核心：把标签和房间轮廓配对 #

这里用点在多边形内部的算法来判断标签属于哪个房间，这是最通用的方法，不管图层和块结构都能用。可以用shapely库简化判断逻辑（先pip install shapely装好）：

from shapely.geometry import Point, Polygon

# 先把每个房间轮廓转换成shapely的Polygon对象，方便后续判断
room_polygons = []
for outline in room_outlines:
    # 提取多段线的所有顶点坐标（只保留XY）
    vertices = [(p[0], p[1]) for p in outline.points]
    poly = Polygon(vertices)
    room_polygons.append({
        "polygon": poly,
        "raw_vertices": vertices,  # 保存原始顶点坐标，方便后续输出
        "bounding_box": poly.bounds  # 也可以直接存外接矩形的坐标（minx, miny, maxx, maxy）
    })

# 开始匹配标签和房间
final_room_data = []
for label in room_labels:
    label_point = Point(label["insert_pos"])
    for room in room_polygons:
        if room["polygon"].contains(label_point):
            final_room_data.append({
                "room_name": label["label_text"],
                "room_vertices": room["raw_vertices"],
                "room_bbox": room["bounding_box"]
            })
            break  # 一个标签对应一个房间，找到就跳出循环

3. 处理特殊场景 #

如果你的CAD图用了**块（Block）**来封装房间（比如一个块里同时包含轮廓和标签），那可以直接遍历块来提取：

for block in dxf.blocks:
    # 跳过CAD自带的系统块（名字带*）
    if block.name.startswith('*'):
        continue
    # 从块里提取轮廓和标签
    block_outlines = [e for e in block.entities if e.dxftype == 'LWPOLYLINE' and e.closed]
    block_labels = [e.text for e in block.entities if e.dxftype in ['TEXT', 'MTEXT']]
    # 这里可以直接把块内的轮廓和标签关联，逻辑更简单

三、备选工具推荐 #

如果dxfgrabber功能不够用，还可以试试这些工具：

• ezdxf：比dxfgrabber功能更全，文档也更详细，支持修改DXF，关联逻辑的实现思路和上面一致
• CADQuery：偏向参数化CAD处理，但提取数据也很顺手，适合结构复杂的CAD图
• Autodesk Forge API：如果是大量DWG文件，可以用官方API直接解析，不用转DXF，但需要申请API密钥

四、调试小技巧 #

1. 先用CAD软件打开DXF，查看目标对象的属性：比如轮廓在哪个图层，标签的插入点位置，有没有属于同一个块——这些信息能帮你快速确定关联规则，少走弯路
2. 调试时先拿一个只有1-2个房间的简单DWG测试，确认逻辑没问题再批量处理
3. 先打印出所有entity的dxftype和属性，看看你的文件里实际用的是哪种对象，别想当然用LWPOLYLINE，有些老CAD可能用POLYLINE

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