11

是否有 Python 函数“outer-zip”,它是zip每个迭代的不同默认值的扩展?

a = [1, 2, 3]   # associate a default value 0
b = [4, 5, 6, 7] # associate b default value 1

zip(a,b)  # [(1, 4), (2, 5), (3, 6)]

outerzip((a, 0), (b, 1)) = [(1, 4), (2, 5), (3, 6), (0, 7)]
outerzip((b, 0), (a, 1)) = [(4, 1), (5, 2), (6, 3), (7, 1)]

我几乎可以使用map复制这个 outerzip 函数,但None作为唯一的默认值:

map(None, a, b) # [(1, 4), (2, 5), (3, 6), (None, 7)]

注意1:内置zip函数可以接受任意数量的迭代,outerzip函数也应该如此。(例如,应该能够与和outerzip((a,0),(a,0),(b,1))类似地计算。)zip(a,a,b)map(None, a, a, b)

注意2:我说的是“outer-zip”,以这个haskell question的风格,但也许这不是正确的术语。

4

3 回答 3

11

它被称为izip_longestzip_longest在 python-3.x 中):

>>> from itertools import zip_longest
>>> a = [1,2,3]
>>> b = [4,5,6,7]
>>> list(zip_longest(a, b, fillvalue=0))
[(1, 4), (2, 5), (3, 6), (0, 7)]
于 2012-10-26T11:14:57.323 回答
4

您可以修改zip_longest以支持您的一般迭代用例。

from itertools import chain, repeat

class OuterZipStopIteration(Exception):
    pass

def outer_zip(*args):
    count = len(args) - 1

    def sentinel(default):
        nonlocal count
        if not count:
            raise OuterZipStopIteration
        count -= 1
        yield default

    iters = [chain(p, sentinel(default), repeat(default)) for p, default in args]
    try:
        while iters:
            yield tuple(map(next, iters))
    except OuterZipStopIteration:
        pass


print(list(outer_zip( ("abcd", '!'), 
                      ("ef", '@'), 
                      (map(int, '345'), '$') )))
于 2012-10-26T11:46:02.023 回答
2

这个函数可以通过扩展每个输入的列表和压缩来定义:

def outerzip(*args):
    # args = (a, default_a), (b, default_b), ...
    max_length = max( map( lambda s: len(s[0]), args))
    extended_args = [ s[0] + [s[1]]*(max_length-len(s[0])) for s in args ]
    return zip(*extended_args)

outerzip((a, 0), (b, 1)) # [(1, 4), (2, 5), (3, 6), (0, 7)]
于 2012-10-26T11:28:31.767 回答