python - 如何迭代多个不同长度的列表，但重复较短列表的最后一个值，直到完成最长列表？

Question

在我的 Python 3 脚本中，我试图根据输入组合来自三个不同列表的三个数字。如果列表大小相同，则 zip 没有问题。但是，我希望能够为特定列表输入单个数字，并且脚本可以重复该数字，直到最长的列表完成。这可以通过 zip_longest 完成。但是，使用 fillvalue 不可能为单独的列表使用单独的填充值。
以这个简单的脚本为例:

from itertools import zip_longest

list1=[1]
list2=[4, 5, 6, 7, 8, 9]
list3=[2]
for l1, l2, l3 in zip_longest(list1, list2, list3):
     print(l1, l2, l3)

这是实际结果:

# 1    4 2
# None 5 None                                                        
# None 6 None                                                         
# None 7 None
# None 8 None
# None 9 None  

这将是我想要的结果:

# 1 4 2
# 1 5 2                                                        
# 1 6 2                                                         
# 1 7 2
# 1 8 2
# 1 9 2                                                        
 

我已经通过手动创建不同的 for 循环并询问列表是否为常量来完成此特定任务，但是 zip_longest 与我所需要的非常接近，我想知道我是否遗漏了一些明显的东西。

Answer 1

您可以使用逻辑 or 运算符来使用较短列表的最后一个元素:

from itertools import zip_longest
list1 = [1]
list2 = ["a", "b", "c", "d", "e", "f"]
list3 = [2]
for l1, l2, l3 in zip_longest(list1, list2, list3):
    print(l1 or list1[-1], l2, l3 or list3[-1])

出去:

1 a 2
1 b 2
1 c 2
1 d 2
1 e 2
1 f 2