Question

我为我的类创建了一个基于数组的迭代器类。可以把它想象成一个普通的数组，其中的函数将元素插入到数组中。到目前为止，我只遇到了链接列表的迭代器，但没有找到适用于普通数组的运气。

class ArrayIterator:

    def __init__(self,array):
        self.i = 0

    def __next__(self):
        first = array[0]
        self.i+=1
        return i

    def __iter__(self):
        return self

Answer 1

检查一下：

class ArrayIterator:
    def __init__(self, array):
        self.i = 0
        self.array = array

    def __next__(self):
        try:
            result = self.array[self.i]
        except IndexError:
            raise StopIteration
        else:
            self.i += 1
            return result

    def __iter__(self):
        return self


my_iter = ArrayIterator([1, 2, 3, 4, 5])

for x in my_iter:
    print(x)

迭代python中的数组

1 个答案: