如果出现IndexError,有没有办法判断一行上的哪个对象是“超出范围”?
考虑以下代码:
a = [1,2,3]
b = [1,2,3]
x, y = get_values_from_somewhere()
try:
a[x] = b[y]
except IndexError as e:
....
如果x
或y
太大而且IndexError
被抓住了,我想知道a
或b
中的哪一个超出范围(所以我可以在except
区块中执行不同的操作)。
显然,我可以将x
和y
分别与len(a)
和len(b)
进行比较,但我很好奇是否还有其他方法可以使用IndexError
。
一个更健壮的方法是利用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
有一种方法,但我不认为它非常强大。错误消息有细微差别:
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,这些不同的消息确实是两个错误之间唯一的区别。
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
包含了引发它的关键。
也许这样的事情?
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')
不,没有办法使用抛出的IndexError来推断出哪两个超出界限。