Python迭代器如何处理异常情况 - 问答

在Python中，迭代器可以通过实现__iter__()和__next__()方法来处理异常情况。当迭代器遇到异常时，它可以选择引发异常或返回一个特定的值来表示异常情况。以下是一个简单的示例，展示了如何在迭代器中处理异常：

class MyIterator:
    def __init__(self, data):
        self.data = data
        self.index = 0

    def __iter__(self):
        return self

    def __next__(self):
        try:
            value = self.data[self.index]
            self.index += 1
            return value
        except IndexError:
            raise StopIteration("迭代完成")
        except Exception as e:
            print(f"发生异常: {e}")
            raise StopIteration("迭代完成")

# 示例数据
data = [1, 2, 3, "a", 5]

# 创建迭代器
iterator = MyIterator(data)

# 使用迭代器
for value in iterator:
    print(value)

在这个示例中，我们定义了一个名为MyIterator的迭代器类。在__next__()方法中，我们使用try-except语句来捕获可能发生的异常。如果发生IndexError，我们引发StopIteration异常来表示迭代完成。对于其他类型的异常，我们打印异常信息并引发StopIteration异常。这样，当迭代器遇到异常时，它可以优雅地处理异常并通知调用者迭代已完成。

0 赞

0 踩