如何处理嵌套循环中的空列表？

Question

我正在寻找一种迭代两个列表的产品的方法，这些列表有时可以是空的。 本来要使用itertools.product

import itertools

list1 = [1, 2]
list2 = ['a', 'b']

for i, j in itertools.product(list1, list2):
    print(i, j)

。

1 a
1 b
2 a
2 b

我 但是，当其中一个列表为空时，循环不会打印任何东西：

list1 = [1, 2]
list2 = []
    
for i, j in itertools.product(list1, list2):
    print(i, j)

而我希望它的行为好像只有一个列表。因此，使用Itertools，请执行相当于：

for i in itertools.product(list1):
    print(i)

它可以返回的等同：

(1,)
(2,)

我可以将其放入长时间的IF语句中，但是我正在寻找一个简单的调整，如果列表数量增加，可以轻松扩展。感谢您的帮助。

Answer 1

将它们放在一个变量中并过滤它们以仅使用非空变量：

import itertools

lists = [1, 2], []

for i in itertools.product(*filter(bool, lists)):
    print(i)

或者如果将它们放在单独的变量中确实是您想要的：

import itertools

list1 = [1, 2]
list2 = []

for i in itertools.product(*filter(bool, (list1, list2))):
    print(i)

两者的输出：

(1,)
(2,)

Answer 2

一种方法是检查任何列表是否为空列表并提高了一定的价值，但这不是最优雅的解决方案。例如：

import itertools

list1 = [1, 2]
list2 = []

if len(list2) == 0:
    list2.append("")

for i, j in itertools.product(list1, list2):
    print(i, j)