python - 在 python 中重构生成器的第一次迭代

Question

我想知道是否有办法重构以下代码

 first_run = True
 for i in gen:
        if first_run:
            last_head = i[1]
            last_tail = i[2]
            last_chrom = i[0]
            first_run = False
        else:
            func(i[1], last_head)
            func(i[1], last_tail)
            last_head = i[1]
            last_tail = i[2]
            last_chrom = i[0]

Answer 1

循环的要点似乎是对可迭代对象的成对连续元素执行一些操作。所以我会查看函数 pairwise，其定义在 itertools module documentation 中给出。 :

def pairwise(iterable):
    "s -> (s0,s1), (s1,s2), (s2, s3), ..."
    a, b = tee(iterable)
    next(b, None)
    return izip(a, b)

请注意，这不是一个实际的 itertools 函数，您必须将实现复制并粘贴到您的代码中。不管怎样，有了这个函数，你的循环可以像这样实现:

for a, b in pairwise(gen):
    func(b[1], a[1])
    func(b[1], a[2])