Python 最佳实践 xrange yield

Question

def list_step(max = 268):    
    """ split in steps of 120 elements at time
    """
    if max > 120:
        for i in xrange(0,max,120):
            if i <= max:
                rest =  max % xrange(0,max,120)[-1]
                if rest != 120:
                    yield(i, rest)
                else:
                    yield(i, 0)
    else:
        yield(max)


a = list_step()
a.next()      return > (0,28) (120,28),ecc

是否可以 return 执行最后一个 next() 时的其余部分，而不是带有其余部分的元组？

这样 :

    a.next()      return > (0) (120),ecc.. (28)

Answer 1

是这样的吗？

def list_step(max = 268):    
    """ split in steps of 120 elements at time
    """
    if max > 120:
        rest_list = [];
        for i in xrange(0,max,120):
            if i <= max:
                rest =  max % xrange(0,max,120)[-1]
                if rest != 120:
                    yield(i)
                    rest_list.append(rest)
                else:
                    yield(i)
                    rest_list.append(0)
        for i in rest_list:
            yield(i)
    else:
        yield(max)


a = list_step()
for i in a: print i;

Answer 2

您可以使用 itertools.chain 将迭代器链接在一起 (documentation)。如果你只是想要一个单一的值 "appended" 到你的生成器，你可以使用它（注意你需要以某种方式将一个单一的项目变成一个可迭代的）。

另外你的 max % xrange(0, max, 120)[-1] 将永远是 max % 120，因为 xrange(0, max, 120) 是最大值，它是 120 的倍数，小于最大值，所以除以它会得到与除以 120（取模）相同的结果。

import itertools

itertools.chain(xrange(0,max,120), [max % 120])

Python 最佳实践 xrange yield

Python best practices xrange yield

python

yield

range

xrange