Python自動提取列表索引並追加數據

Question

我不知道如何描述問題。 舉個例子：

a=[[1,2],[3,4],[5,6]]
b=[['a','b'],['c','c']]
x=[a,b]

現在，我想將x元素的元素附加到之前的元素（在這種情況下， b元素在每個a元素上），我可以使用

t=[]
for i in a:
    for j in b:
        t.append(i+j)

然后我想要的結果變成了：

t
[[1, 2, 'a', 'b'],
 [1, 2, 'c', 'c'],
 [3, 4, 'a', 'b'],
 [3, 4, 'c', 'c'],
 [5, 6, 'a', 'b'],
 [5, 6, 'c', 'c']]

在這種情況下，我知道x中有a和b ，因此可以附加它們。 但是，如果我不知道x中有多少個項目，該如何追加元素？

像x=[a,b,c,d,e,...]

嘗試過使用循環，但效果不好。 我在考慮combination但不確定如何做到這一點。

Answer 1

生成成對產品，並分別使用itertools的product和chain功能對其進行展平：

from itertools import chain, product    
t = [list(chain.from_iterable(i)) for i in product(a, b)]

print(t)
[[1, 2, 'a', 'b'],
 [1, 2, 'c', 'c'],
 [3, 4, 'a', 'b'],
 [3, 4, 'c', 'c'],
 [5, 6, 'a', 'b'],
 [5, 6, 'c', 'c']]

該解決方案將推廣到任意數量的列表：

x = [a, b, c, ...]
t = [list(chain.from_iterable(i)) for i in product(*x)]