Question

我注意到itertools文档中的Recipes部分有一些有用的代码，例如consume，但我发现了一个问题。

from itertools import islice
numbers = iter(range(10))
for i in numbers:
    print i 
    next(islice(numbers, 3, 3), None)

上面的代码工作正常，打印[0,4,8]，next(islice(numbers, 3, 3), None)诀窍来自consume食谱。

我们可以看到islice在这里发挥了重要作用。但是，当我用该函数的python文档中提供的代码替换itertools.islice时，代码无法正常工作：

#from itertools import islice

def islice(iterable, *args):
    # islice('ABCDEFG', 2) --> A B
    # islice('ABCDEFG', 2, 4) --> C D
    # islice('ABCDEFG', 2, None) --> C D E F G
    # islice('ABCDEFG', 0, None, 2) --> A C E G
    s = slice(*args)
    it = iter(xrange(s.start or 0, s.stop or 200, s.step or 1))
    nexti = next(it)
    for i, element in enumerate(iterable):
        if i == nexti:
            yield element
            nexti = next(it)


numbers = iter(range(10))
for i in numbers:
    print i 
    next(islice(numbers, 3, 3), None)

输出与上面的输出不同;打印范围内的所有数字，islice()消费行似乎没有做任何事情。

任何人都可以解释为什么会这样，islice在这里有效吗？

Answer 1

切片的开始和结束参数相等，因此对xrange的next(it)调用会引发StopIteration：

>>> s = slice(3, 3, None)
>>> it = iter(xrange(s.start or 0, s.stop or 200, s.step or 1))
>>> next(it)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
StopIteration

并且slice()生成器函数立即终止。这是合乎逻辑的，xrange()函数在这种情况下不会产生任何数字。

itertools文档是说明性，并不意味着完整。您已经找到了一个边缘例，其中示例Python代码的工作方式与C实现的工作方式不同。

为了使Python代码也适用于这种情况，请捕获StopIteration异常，并在循环中显式测试edgecase：

def islice(iterable, *args):
    s = slice(*args)
    it = iter(xrange(s.start or 0, s.stop or 200, s.step or 1))
    try:
        nexti = next(it)
        consume = False
    except StopIteration:
        if s.start < 1:
            return
        nexti = s.start - 1
        consume = True
    for i, element in enumerate(iterable):
        if i == nexti:
            if consume:
                return
            yield element
            nexti = next(it)

Python示例代码使用slice()对象使代码更紧凑，更易于阅读; C代码执行自己的启动，停止和步骤解析，并处理计数本身而不是委托给xrange()，因此不会遇到此边缘情况。

为什么itertools.islice的Python版本不起作用？

1 个答案: