Python:创建一个可以循环迭代的类

2024-10-03 17:19:40 发布

您现在位置:Python中文网/ 问答频道 /正文

我想在Python中创建一个类似于列表的类,但可以循环迭代 用例示例:

myc = SimpleCircle()
print(len(myc))
j = iter(myc)
for i in range (0, 5):
    print(next(j))

它会打印出来 一 b c d a

到目前为止我尝试的代码是下面的代码 我知道问题出在我的__next__

方法 顺便说一下,这似乎被忽略了,我可以使用next,即使我没有实现它

class SimpleCircle:
    def __init__(self):
        self._circle = ['a', 'b', 'c', 'd']
        self._l = iter(self._circle)


    def __len__(self):
        return len(self._circle)

    def __iter__(self):
        return (elem for elem in self._circle)

    def __next__(self):
        try:
            elem = next(self._l)
            idx = self._circle.index(elem)
            if idx < len(self._circle):
                return elem
            else:
                return self._circle[0]
        except StopIteration:
            pass

Tags: 代码inselfforlenreturndefnext
2条回答

下面是一个基本的非itertools实现:

class CyclicIterable:
    def __init__(self, data):
        self._data = list(data)

    def __iter__(self):
        while True:
            yield from self._data

cycle = CyclicIterable(['a', 'b', 'c', 'd'])
for i, x in zip(range(5), cycle):
    print(x)

注意,没有必要实现__next__,因为Cycle类本身与list一样,不是迭代器。要显式地从中获取迭代器,可以编写:

it = cycle.__iter__()
print(next(it))
print(next(it))
print(next(it))
print(next(it))
print(next(it))

当然,您可以实例化任意多个迭代器。你知道吗

这实际上已经存在于^{}中,例如:

from itertools import cycle

for x in cycle(['a', 'b', 'c', 'd']):
    print(x)

将不断重复元素。你知道吗

接下来,您将把iterable和迭代器混合起来,它们通常是不同的东西。你知道吗

作为一个iterable,我们可以继续从self._circle迭代:

class SimpleCircle:
    def __init__(self):
        self._circle = ['a', 'b', 'c', 'd']

    def __len__(self):
        return len(self._circle)

    def __iter__(self):
        if not self._circle:
            raise StopIteration
        while True:
            yield from self._circle

或者对于迭代器:

class CycleIterator:

    def __init__(self, iterable):
        self.iterator = iter(iterable)
        self.__next__ = self._iternext
        self.idx = 0
        self.list = []

    def _iternext(self):
        try:
            x = next(self.iterator)
            self.list.append(x)
            return x
        except StopIteration:
            self.__next__ = self._iterlist
            return self._iterlist()

    def _iterlist(self):
        try:
            return self.list[self.index]
        except IndexError:
            raise StopIteration
        finally:
            self.index = (self.index + 1) % len(self.list)

相关问题 更多 >