Python单表达式多列表触发IndexError时如何快速定位出错列表

可以通过两种思路实现无IDE依赖的快速定位，分别是轻量代码改造的包装函数法，和无侵入的异常栈解析法：

方案1：自定义列表取值包装函数（改造成本极低，可读性高） #

通过一个简单的包装函数封装列表取值逻辑，出错时直接输出对应的列表标识、索引和长度，不需要额外依赖。

def get(lst: list, idx: int, list_name: str):
    try:
        return lst[idx]
    except IndexError:
        raise IndexError(f"列表{list_name}索引{idx}越界，当前列表长度为{len(lst)}") from None

使用方式 #

把原代码中的列表取值都替换为get调用即可：

• 平级表达式改造示例：
  原代码 a[1] = b[2] + c[3] + d[4] 改为：

  a[1] = get(b, 2, 'b') + get(c, 3, 'c') + get(d, 4, 'd')
  

• 嵌套索引改造示例：
  原代码 a[1] = b[c[d[1]]] 改为：

  a[1] = get(b, get(c, get(d, 1, 'd'), 'c'), 'b')
  

运行后一旦触发越界，会直接输出类似IndexError: 列表d索引1越界，当前列表长度为1的报错，一眼就能定位问题。

方案2：无侵入异常栈解析（不需要修改原有表达式） #

如果不想修改原有代码的写法，可以在try/except块中通过inspect模块解析出错位置的代码和运行时变量，自动定位出错的列表。

import inspect
import re

# 原有业务代码不需要任何修改
try:
    a[1] = b[c[d[1]]]
except IndexError:
    # 获取当前执行栈帧
    frame = inspect.currentframe().f_back
    # 读取出错行的源代码
    line_code = inspect.getframeinfo(frame).code_context[0].strip()
    # 匹配所有「变量名[索引]」格式的列表取值操作
    matches = re.findall(r'(\w+)\[(.+?)\]', line_code)
    # 逐个校验每个列表的索引是否合法
    for list_name, idx_expr in matches:
        # 从局部/全局变量中拿到列表实例
        lst = frame.f_locals.get(list_name, frame.f_globals.get(list_name))
        if not isinstance(lst, list):
            continue
        try:
            # 计算索引的实际取值
            idx = eval(idx_expr, frame.f_globals, frame.f_locals)
            # 校验索引是否越界
            if idx >= len(lst) or idx < -len(lst):
                print(f"定位到越界列表：{list_name}，索引值：{idx}，列表长度：{len(lst)}")
                break
        except:
            continue
    # 继续抛出原有异常
    raise

该方案适用于不想调整原有业务代码结构的场景，普通复杂度的表达式都可以自动识别定位。

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