python - Python 最佳实践 xrange 产量

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

是否有可能在执行最后一个 next() 时返回其余部分，而不是返回包含其余部分的元组？

以便 ：

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

Answer 1

您可以使用itertools.chain将迭代器链接在一起（文档）。如果您只是想将单个值“附加”到您的生成器中，您可以使用它（请注意，您需要以某种方式将单个项目转换为可迭代项）。

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

import itertools

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

Answer 2

像这样的东西？

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;