Python循环向列表插入元素问题：分页脚本报错修正求助

需求：调用createFooter2(34, 65, 3, 3)时，生成字符串："1 2 3 ... 31 32 33 34 35 36 37 ... 63 64 65"，现有代码在插入省略号时存在问题，需修改。

现有问题代码：

def createFooter2(current_page, total_pages, boundaries, around) -> str:
    footer = []
    page = 1
    #Append lower boundaries
    while page <= boundaries:
        footer.append(page)
        page += 1
    #Append current page and arround
    page = current_page - around
    while page <= current_page + around:
        footer.append(page)
        page += 1
    #Append upper boundaries
    page = total_pages - boundaries + 1
    while page <= total_pages:
        footer.append(page)
        page += 1
    #Add Ellipsis if necessary
    for i in range(len(footer)):
        if i > 0 and footer[i] - footer[i - 1] > 1:
            footer.insert(i, "...")
    result = ' '.join(str(page) for page in result)
    print(result)
    return result

createFooter2(34, 65, 3, 3)

问题分析与修改方案 #

核心问题 #

1. 遍历插入元素导致索引混乱：正向遍历列表时插入元素，会改变列表长度，后续循环的索引会错位，导致重复插入或遗漏。
2. 未过滤无效页码：current_page周围的页码可能小于1或大于total_pages，直接加入会出现非法页码。
3. 存在重复页码：当boundaries范围和current_page周围范围重叠时，会出现重复页码。
4. 变量名错误：最后拼接时用了未定义的result变量，应该用footer。

修改后的代码 #

def createFooter2(current_page, total_pages, boundaries, around) -> str:
    # 收集所有需要显示的页码，用集合去重
    pages_set = set()
    
    # 添加下边界页码
    for page in range(1, boundaries + 1):
        pages_set.add(page)
    
    # 添加当前页周围的有效页码（过滤超出范围的）
    start = max(1, current_page - around)
    end = min(total_pages, current_page + around)
    for page in range(start, end + 1):
        pages_set.add(page)
    
    # 添加上边界页码
    for page in range(total_pages - boundaries + 1, total_pages + 1):
        pages_set.add(page)
    
    # 转成有序列表
    sorted_pages = sorted(pages_set)
    
    # 构建最终导航字符串，处理省略号
    footer = []
    for i in range(len(sorted_pages)):
        current = sorted_pages[i]
        if i == 0:
            footer.append(str(current))
        else:
            prev = sorted_pages[i-1]
            if current - prev > 1:
                footer.append("...")
            footer.append(str(current))
    
    result = ' '.join(footer)
    print(result)
    return result

createFooter2(34, 65, 3, 3)

关键修改说明 #

1. 用集合去重：避免重复添加重叠范围的页码。
2. 过滤无效页码：用max(1, ...)和min(total_pages, ...)确保只添加1到total_pages之间的页码。
3. 有序列表处理省略号：先把页码排序，再遍历，通过比较当前页和前一页的差值判断是否需要插入省略号，避免遍历插入的索引问题。
4. 修复变量名错误：拼接时使用正确的变量。

运行修改后的代码，会输出预期的：1 2 3 ... 31 32 33 34 35 36 37 ... 63 64 65

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