QGIS图层自动分组Python脚本实现

1. 为什么需要图层自动分组功能

在QGIS项目中处理大量同名图层时，手动管理往往效率低下且容易出错。想象一下，当你从不同数据源加载了十几个名为"道路"或"建筑物"的图层时，项目面板很快就会变得杂乱无章。这种情况在以下场景尤为常见：

• 处理多期遥感影像数据时，每期影像可能使用相同的命名规范
• 整合来自不同部门的行政区划数据，各部门可能采用相同的图层命名
• 使用自动化脚本批量导入数据时生成的同名临时图层

传统的手动分组方式需要逐个右键点击图层，选择"添加到组"并命名组别。对于20个以上的同名图层，这个过程可能耗费10-15分钟，且容易遗漏或误操作。更糟糕的是，当需要更新数据重新加载图层时，所有分组工作又得重来一遍。

2. QGIS原生分组功能的局限性

QGIS默认提供的图层分组功能虽然基础可用，但在处理批量图层时存在明显不足：

2.1 手动操作效率低下

每次只能选择一个图层添加到组，无法批量选择同名图层。即使使用Ctrl键多选，也只能将选中的图层加入现有组，不能根据名称自动创建新组。

2.2 缺乏智能匹配

原生功能无法识别图层名称中的共同特征（如相同前缀、后缀或命名模式），导致用户必须肉眼识别相关图层再进行手动分组。

2.3 无法保存分组规则

每次新建项目或重新加载数据时，分组结构不会自动保留，需要重复劳动。这对于长期项目或需要频繁更新数据的用户尤为不便。

3. 自动分组解决方案的核心思路

针对上述痛点，我们可以通过Python脚本扩展QGIS的功能，实现智能化的图层自动分组。这个方案的核心逻辑包含三个关键部分：

3.1 图层名称分析器

python复制def analyze_layer_names(layers):
    name_dict = {}
    for layer in layers:
        base_name = layer.name().split('_')[0]  # 假设使用下划线分隔命名
        if base_name not in name_dict:
            name_dict[base_name] = []
        name_dict[base_name].append(layer)
    return name_dict

这段代码会将所有图层按名称的前缀部分（下划线前的文字）进行分类。例如"道路_2020"和"道路_2021"会被归入同一组。

3.2 动态组创建器

python复制def create_group(group_name, parent=None):
    root = QgsProject.instance().layerTreeRoot()
    return root.addGroup(group_name) if parent is None else parent.addGroup(group_name)

该函数负责在指定位置（根目录或已有组内）创建新组，确保组结构可以嵌套多层。

3.3 智能分配算法

python复制def auto_group_layers():
    layers = QgsProject.instance().mapLayers().values()
    name_groups = analyze_layer_names(layers)
    
    for group_name, layer_list in name_groups.items():
        if len(layer_list) > 1:  # 只有同名图层数量大于1时才创建组
            group = create_group(group_name)
            for layer in layer_list:
                QgsProject.instance().layerTreeRoot().findLayer(layer.id()).setParent(group)

这个主函数将所有图层按名称分组后，自动创建对应的组并将图层移入其中。

4. 完整实现步骤详解

4.1 准备Python控制台

在QGIS中通过顶部菜单"插件"→"Python控制台"打开交互式Python环境。点击工具栏上的"显示编辑器"按钮，可以打开更强大的脚本编辑器。

4.2 创建自动分组脚本

将以下完整脚本复制到编辑器中：

python复制from qgis.core import QgsProject

def auto_group_layers(separator='_', min_group_size=2):
    """
    自动将同名图层分组
    :param separator: 名称分隔符，默认为下划线
    :param min_group_size: 最小分组尺寸，默认为2（即至少2个同名图层才创建组）
    """
    project = QgsProject.instance()
    root = project.layerTreeRoot()
    layers = project.mapLayers().values()
    
    # 统计图层名称分布
    name_dict = {}
    for layer in layers:
        parts = layer.name().split(separator)
        base_name = parts[0] if len(parts) > 1 else layer.name()
        
        if base_name not in name_dict:
            name_dict[base_name] = []
        name_dict[base_name].append(layer)
    
    # 创建分组并移动图层
    for group_name, layer_list in name_dict.items():
        if len(layer_list) >= min_group_size:
            group_node = root.addGroup(group_name)
            for layer in layer_list:
                layer_node = root.findLayer(layer.id())
                if layer_node:
                    layer_node.clone().setParent(group_node)
                    root.removeChildNode(layer_node)

# 执行自动分组（使用默认参数）
auto_group_layers()

4.3 自定义分组规则

脚本提供了两个关键参数供调整：

• separator: 修改此参数可以改变名称分割方式。例如设为'-'则按连字符分组
• min_group_size: 设置创建组的最小图层数量，避免单个图层也生成组

4.4 创建快捷按钮（可选）

对于需要频繁使用此功能的用户，可以将脚本保存为.py文件，然后通过以下步骤创建工具栏按钮：

1. 打开"设置"→"自定义"
2. 切换到"工具栏"标签
3. 新建工具栏（如"My Tools"）
4. 切换到"命令"标签，添加"Python脚本"命令
5. 指定脚本路径并设置图标
6. 将新命令拖到刚创建的工具栏上

5. 高级应用技巧

5.1 处理嵌套分组结构

对于需要多级分组的场景（如"中国/浙江省/杭州市"），可以修改脚本支持层级分组：

python复制def auto_nested_group(separators=('_', '/')):
    # 实现代码与之前类似，但支持多级分隔符
    pass

5.2 正则表达式匹配

更复杂的命名规则可以使用正则表达式提取组名：

python复制import re

def extract_group_name(layer_name):
    match = re.match(r'(Road|Building)_(\d{4})', layer_name)
    return match.group(1) if match else None

5.3 保存分组模板

通过以下代码可以将当前分组结构导出为模板，供其他项目使用：

python复制def export_group_structure():
    groups = {}
    root = QgsProject.instance().layerTreeRoot()
    # 递归收集分组信息
    # ...
    return groups

6. 常见问题解决方案

6.1 脚本执行无反应

• 检查Python控制台是否有错误输出
• 确认QGIS项目中有多个同名图层
• 尝试降低min_group_size参数值

6.2 分组后图层顺序混乱

QGIS默认按字母顺序排列组内图层。如需保持原顺序，可以在移动图层时记录原始位置：

python复制original_order = {layer.id(): idx for idx, layer in enumerate(layers)}
# 移动图层后按original_order重新排序

6.3 处理特殊字符

图层名称包含特殊字符时可能导致脚本异常。添加以下预处理：

python复制safe_name = re.sub(r'[^\w-]', '', raw_name)

7. 性能优化建议

当处理超大型项目（100+图层）时，可以考虑以下优化：

7.1 增量分组

python复制def incremental_group(new_layers):
    # 只对新添加的图层进行处理
    pass

7.2 后台执行

使用QgsTask将分组操作放在后台执行，避免界面卡顿：

python复制from qgis.core import QgsTask

class GroupingTask(QgsTask):
    def run(self):
        auto_group_layers()
        return True

7.3 缓存图层树状态

在操作前保存图层可见状态，分组后恢复：

python复制def save_visibility():
    return {layer.id(): layer.isVisible() for layer in layers}