大型表格下wx.grid.AutoSizeColumn性能问题及优化方案咨询

问题描述 #

使用wx.grid.GridTableBase填充了一个约9400行×14列的大型表格后，执行以下代码进行列宽自动调整：

[self.grid.AutoSizeColumn(col) for col in range(self.grid.GetNumberCols())]

经cProfile分析该操作耗时26秒，无法满足应用性能要求，需寻找耗时低于1秒的替代方案。

高效替代方案 #

1. 仅计算可见行+表头的最大宽度 #

AutoSizeColumn默认会遍历所有行计算宽度，这是耗时的核心原因。可以只遍历当前可见的行（加上表头）来计算最大宽度，大幅减少计算量：

def fast_autosize_columns(grid):
    grid.BeginBatch()
    try:
        for col in range(grid.GetNumberCols()):
            # 获取表头宽度
            header_width = grid.GetTextExtent(grid.GetColLabelValue(col)).width
            max_width = header_width
            # 获取可见行范围
            start_row = grid.GetVisibleRowsBegin()
            end_row = grid.GetVisibleRowsEnd()
            # 遍历可见行计算单元格宽度
            for row in range(start_row, end_row):
                cell_value = grid.GetCellValue(row, col)
                cell_width = grid.GetTextExtent(cell_value).width
                if cell_width > max_width:
                    max_width = cell_width
            # 设置列宽，额外加5像素留白
            grid.SetColSize(col, max_width + 5)
    finally:
        grid.EndBatch()

调用方式：fast_autosize_columns(self.grid)
这个方法只处理可见行，在大多数场景下能把耗时压缩到0.1秒以内，完全满足要求。

2. 预计算表格数据的最大宽度 #

如果你的表格数据是从数据源（比如列表、数据库）加载的，可以在填充表格前就计算每列的最大宽度，避免在UI层重复遍历：

# 假设data是二维列表，对应表格数据
def calculate_col_max_widths(data, grid):
    max_widths = []
    # 处理表头
    headers = [grid.GetColLabelValue(col) for col in range(len(data[0]))]
    for header in headers:
        max_widths.append(grid.GetTextExtent(header).width)
    # 处理数据行
    for row in data:
        for col_idx, cell_value in enumerate(row):
            width = grid.GetTextExtent(str(cell_value)).width
            if width > max_widths[col_idx]:
                max_widths[col_idx] = width
    return max_widths

# 填充表格后直接设置列宽
max_widths = calculate_col_max_widths(your_data_source, self.grid)
self.grid.BeginBatch()
try:
    for col, width in enumerate(max_widths):
        self.grid.SetColSize(col, width + 5)
finally:
    self.grid.EndBatch()

这种方法完全绕开了wxGrid的行遍历，直接从数据源计算，耗时通常在0.5秒以内（取决于数据量）。

3. 必加的重绘优化 #

在调整列宽前后分别调用BeginBatch()和EndBatch()，避免每调整一列就触发一次重绘，减少UI层面的耗时叠加——上面两个方案已经包含了这个优化。

核心优化逻辑 #

AutoSizeColumn慢的根源是遍历所有9400行进行UI层面的文本宽度计算，只要砍掉不必要的遍历（只处理可见行）或者把计算移到数据层，就能把耗时降到1秒以内。

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