基于xlwings的Excel数据处理脚本需求及代码优化问询

需求回顾 #

需要实现三个核心功能：

• 在A列姓名中做部分字符串匹配（示例中匹配含'DAD'的姓名），同时筛选出C列（y坐标）相同的条目
• 对每组同y坐标的匹配条目，计算B列（x坐标）的最大值和最小值
• 为x最大值对应的A列姓名追加'_start'，x最小值对应的追加'_end'

修复后的完整代码 #

import xlwings as xw

def process_name_coords():
    # 获取当前活动工作簿和指定工作表
    wb = xw.books.active
    sht = wb.sheets['new Sheet']
    
    # 获取A列最后一行的行号（忽略空行）
    last_row = sht.range('A' + str(sht.cells.last_cell.row)).end('up').row
    if last_row < 2:  # 没有有效数据（表头在第1行）
        print("没有可处理的数据")
        return
    
    # 存储分组数据：key是y坐标，value是列表，每个元素是(x坐标, 行号)
    y_groups = {}
    
    # 遍历第2行到最后一行的数据
    for row in range(2, last_row + 1):
        name = sht.range(f'A{row}').value
        x_coord = sht.range(f'B{row}').value
        y_coord = sht.range(f'C{row}').value
        
        # 跳过空值或非数值的坐标
        if not name or not isinstance(x_coord, (int, float)) or not isinstance(y_coord, (int, float)):
            continue
        
        # 检查姓名是否包含目标字符串（这里是'DAD'，可按需修改）
        if 'DAD' in str(name):
            # 按y坐标分组存储x和行号
            if y_coord not in y_groups:
                y_groups[y_coord] = []
            y_groups[y_coord].append((x_coord, row))
    
    # 处理每个y坐标分组
    for y, items in y_groups.items():
        # 跳过只有一个条目的组（没有最大最小的意义）
        if len(items) < 2:
            continue
        
        # 找到x最大和最小的条目
        max_x_item = max(items, key=lambda x: x[0])
        min_x_item = min(items, key=lambda x: x[0])
        
        # 给对应姓名追加标记
        # 处理最大值对应的姓名
        max_row = max_x_item[1]
        original_name = str(sht.range(f'A{max_row}').value)
        if '_start' not in original_name:  # 避免重复追加
            sht.range(f'A{max_row}').value = f"{original_name}_start"
        
        # 处理最小值对应的姓名
        min_row = min_x_item[1]
        original_name = str(sht.range(f'A{min_row}').value)
        if '_end' not in original_name:  # 避免重复追加
            sht.range(f'A{min_row}').value = f"{original_name}_end"

# 执行函数
process_name_coords()

关键逻辑说明 #

1. 数据读取与筛选：
   • 正确获取有效数据范围，跳过空行和无效值
   • 严格检查姓名的部分匹配，同时确保x、y坐标是数值类型，避免后续计算错误
2. 分组存储：
   • 用字典按y坐标分组，每组存储对应的x坐标和所在行号，方便后续定位修改单元格
3. 最大最小值处理：
   • 用max()和min()函数配合lambda快速找到每组的x极值对应的行
   • 增加重复追加检查，避免多次运行时重复添加标记
4. 代码规范：
   • 修复原代码的缩进错误，函数结构更清晰
   • 增加异常情况处理（比如无数据、分组只有一个条目）

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