Zigzag迭代器的Pythonic方法？

Pythoner

我正在编程Zigzag迭代器，它是通过以下方式迭代2D列表的：

[1,4,7]
[2,5,8,9]
[3,6]

至

[1,2,3,4,5,6,7,8,9]

我实现了一个算法：

class ZigzagIterator:

    def __init__(self, vecs):

        self.vecs = []
        self.turns = 0
        for vec in vecs:
            vec and self.vecs.append(iter(vec))

    def next(self):
        try:
            elem = self.vecs[self.turns].next()
            self.turns = (self.turns+1) % len(self.vecs)
            return elem
        except StopIteration:
            self.vecs.pop(self.turns)
            if self.hasNext():
                self.turns %= len(self.vecs)

    def hasNext(self):
        return len(self.vecs) > 0

if __name__ == "__main__":
    s = ZigzagIterator([[1,4,7],[2,5,8,9],[3,6]])
    while s.hasNext():
        print s.next()

>>> 1 2 3 4 5 6 7 8 None None 9 None

我知道问题是因为我在每个列表的next（）上再调用1次，然后得到3个None。我可以通过使用Java检查hasnext方法来解决此问题。我也可以在python中实现hasnext迭代器。我的问题是我该如何以更Python的方式解决此问题，而不是用Java来思考。

亚当·史密斯

这是itertoolsdocs中的循环食谱。

def roundrobin(*iterables):
    "roundrobin('ABC', 'D', 'EF') --> A D E B F C"
    # Recipe credited to George Sakkis
    num_active = len(iterables)
    nexts = cycle(iter(it).__next__ for it in iterables)
    while num_active:
        try:
            for next in nexts:
                yield next()
        except StopIteration:
            # Remove the iterator we just exhausted from the cycle.
            num_active -= 1
            nexts = cycle(islice(nexts, num_active))

本文收集自互联网，转载请注明来源。

如有侵权，请联系 [email protected] 删除。