Python列表内部相邻元素分组报错（IndexError）排查及实现咨询

为什么会出现IndexError？ #

你的代码里用了for x in range(len(inp)-1)来循环，但这个循环的迭代次数是基于初始列表长度计算的。而在循环过程中，你执行了del inp[x+1]操作，这会直接缩短列表的长度。举个具体的例子：

• 初始列表长度是8，range(len(inp)-1)会生成0到6的序列（共7次循环）
• 当你合并前两个元素并删除第二个后，列表长度变成7；继续合并几次后，列表长度会进一步缩短
• 当循环到后面的x值时，x+1会超出当前列表的有效索引范围（比如列表只剩5个元素时，x=4的话x+1=5，但列表最大索引是4），直接触发IndexError

另外你的原函数还有个逻辑漏洞：合并元素后，x会自动递增，导致你跳过了对合并后新元素与下一个元素的检查（比如合并[1]和[2]成[1,2]后，应该继续检查它和[3]是否需要合并，但原循环的x会走到1，直接跳过这个判断）。

修复方案：两种实现思路 #

思路1：修改原函数，用while循环动态控制索引 #

如果一定要在原列表上修改，可以改用while循环，这样能根据列表的实时长度调整循环条件，同时控制x的递增时机：

def is_pos(n):
    return n > 0  # 补全你用到的is_pos函数

def gp(inp):
    x = 0
    while x < len(inp) - 1:
        # 检查当前元素和下一个元素是否都是正数
        if is_pos(inp[x][0]) and is_pos(inp[x+1][0]):
            inp[x] += inp[x+1]
            del inp[x+1]
            # 合并后不递增x，继续检查当前位置的新元素与下一个元素
        else:
            x += 1  # 不需要合并时，再移动到下一个元素

测试你的示例输入：

inp = [[1],[2],[3],[-4],[-3],[-2],[-1],[0]]
gp(inp)
print(inp)  # 输出 [[1,2,3], [-4,-3,-2,-1], [0]]

思路2：创建新列表构建结果（推荐） #

在Python中，直接修改正在遍历的列表很容易引发索引问题，更优雅且安全的方式是创建一个新列表来逐步构建分组结果，不修改原列表：

def is_pos(n):
    return n > 0

def group_adjacent_positives(lst):
    if not lst:  # 处理空列表的边界情况
        return []
    
    # 初始化结果列表，复制第一个元素避免修改原列表
    result = [lst[0].copy()]
    for current_item in lst[1:]:
        last_group = result[-1]
        # 判断当前组和当前元素是否属于同一分组（都是正数）
        if is_pos(last_group[0]) and is_pos(current_item[0]):
            last_group.extend(current_item)
        else:
            # 不属于同一分组，添加新的组到结果
            result.append(current_item.copy())
    return result

调用示例：

original_list = [[1],[2],[3],[-4],[-3],[-2],[-1],[0]]
grouped_list = group_adjacent_positives(original_list)
print(grouped_list)  # 输出 [[1,2,3], [-4,-3,-2,-1], [0]]

这种方式的优势：

• 不会修改原列表，避免了副作用
• 逻辑清晰，更容易理解和维护
• 天然避免了索引越界的问题

通用相邻元素分组的扩展思路 #

如果要把这个逻辑扩展为通用的相邻元素分组（不仅仅是正数），可以把分组判断条件抽成一个参数，让函数更灵活：

def group_adjacent(lst, condition):
    if not lst:
        return []
    result = [lst[0].copy()]
    for item in lst[1:]:
        last = result[-1]
        if condition(last[0], item[0]):
            last.extend(item)
        else:
            result.append(item.copy())
    return result

# 用正数分组的条件调用
def is_pos_pair(a, b):
    return is_pos(a) and is_pos(b)

original_list = [[1],[2],[3],[-4],[-3],[-2],[-1],[0]]
print(group_adjacent(original_list, is_pos_pair))

这样你可以根据不同的需求，自定义condition函数来实现不同的相邻分组逻辑。

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