Outerzip / zip最长的功能(具有多个填充值)
问题内容:
是否有一个Python函数“外部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复制此externalzip函数,但将其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:我以 这种haskell问题的方式说“ outer-zip” ,但这也许不是正确的术语。
问题答案:
您可以修改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'), '$') )))