如何找出哪个指数超出范围?

问题描述 投票:8回答:5

如果出现IndexError,有没有办法判断一行上的哪个对象是“超出范围”?

考虑以下代码:

a = [1,2,3]
b = [1,2,3]

x, y = get_values_from_somewhere()

try:
   a[x] = b[y]
except IndexError as e:
   ....

如果xy太大而且IndexError被抓住了,我想知道ab中的哪一个超出范围(所以我可以在except区块中执行不同的操作)。

显然,我可以将xy分别与len(a)len(b)进行比较,但我很好奇是否还有其他方法可以使用IndexError

python exception index-error
5个回答
4
投票

一个更健壮的方法是利用sys.exc_info()返回的追溯对象,从框架中提取由文件名和行号指示的代码,使用ast.parse解析该行,子类ast.NodeVisitor找到所有Subscript节点,unparse(带有astunparse)并使用框架的全局变量和局部变量来评估节点,以查看哪些节点导致异常,并打印有问题的表达式:

import sys
import linecache
import ast
import astunparse

def find_index_error():
    tb = sys.exc_info()[2]
    frame = tb.tb_frame
    lineno = tb.tb_lineno
    filename = frame.f_code.co_filename
    line = linecache.getline(filename, lineno, frame.f_globals)
    class find_index_error_node(ast.NodeVisitor):
        def visit_Subscript(self, node):
            expr = astunparse.unparse(node).strip()
            try:
                eval(expr, frame.f_globals, frame.f_locals)
            except IndexError:
                print("%s causes IndexError" % expr)
    find_index_error_node().visit(ast.parse(line.strip(), filename))

a = [1,2,3]
b = [1,2,3]
x, y = 1, 2
def f():
    return 3
try:
    a[f() - 1] = b[f() + y] + a[x + 1] # can you guess which of them causes IndexError?
except IndexError:
    find_index_error()

这输出:

b[(f() + y)] causes IndexError

8
投票

有一种方法,但我不认为它非常强大。错误消息有细微差别:

a = [1,2,3]
b = [1,2,3]

a[2] = b[3]

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-69-8e0d280b609d> in <module>()
      2 b = [1,2,3]
      3 
----> 4 a[2] = b[3]

IndexError: list index out of range

但如果错误在左侧:

a[3] = b[2]

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
<ipython-input-68-9d66e07bc70d> in <module>()
      2 b = [1,2,3]
      3 
----> 4 a[3] = b[2]

IndexError: list assignment index out of range

请注意消息中的“赋值”。

所以,你可以这样做:

a = [1,2,3]
b = [1,2,3]

try:
    a[3] = b[2]
except IndexError as e:
    message = e.args[0]
    if 'assignment' in message:
        print("Error on left hand side")
    else:
        print("Error on right hand side")

输出:

# Error on left hand side

同样,我不会太信任它,如果消息在另一个版本的Python中发生变化,它将失败。


我看了一下these parts of the source code,这些不同的消息确实是两个错误之间唯一的区别。


5
投票

IndexError异常不存储有关引发异常的信息。它唯一的数据是错误信息。你必须制作你的代码才能这样做。

a = [1,2,3]
b = [1,2,3]

x, y = get_values_from_somewhere()

try:
    value = b[y]
except IndexError as e:
   ...

try:
    a[x] = value
except IndexError as e:
    ...

我想补充一下,我试图通过inspect.frame恢复罪魁祸首,但无法这样做。因此,我怀疑确实没有强大的方法。

最后,请注意这是IndexError特有的,因为其他异常可能包含推断导致它们的原因所需的信息。举例来说,KeyError包含了引发它的关键。


0
投票

也许这样的事情?

a = [1,2,3]
b = [2,3,4]
x = 5
y = 1

try:
    a[x] = b[y]
except IndexError:
    try:
        a[x]
        print('b caused indexerror')
    except IndexError:
        print('a caused indexerror')

-1
投票

不,没有办法使用抛出的IndexError来推断出哪两个超出界限。

© www.soinside.com 2019 - 2024. All rights reserved.