Python for 循环中捕获异常

Question

我有以下 for 循环：

for batch in loader:
    # do something with batch
    ...

从加载器中提取批处理时，我的循环有时会失败。我想做的是类似于下面的代码片段，但我希望能够继续循环下一个值，而不是跳过其余的值。

error_idxs = [] 

try:
    for i,batch in enumerate(loader):
        # do something with batch
        ...
except:
    error_idxs.append(i)

上述方法的问题在于，一旦发生异常，它就会退出循环，而不是继续下一个批次。

有没有办法在下一批中继续循环？

Answer 1

error_idxs = []
i = 0
while True:
    try:
        batch = next(loader)
        # do something
    except StopIteration:
        break
    except Exception:
        error_idxs.append(i)
    finally:
        i += 1

编辑：更正StopIterationError并StopIteration删除continue