python组合算法的问题

Question

求组合数的问题：

假设我有2个list

A=[a1,a2]
B=[b1,b2]

从这个2个列表中各取一个值，形成一个组合比如[a1,b1]，或者 [a1,b2]这样的，代码这样写:

for a in A:
    for b in B:
        res.append([a,b])

但如果我有3个列表呢，代码这样写：

for a in A:
    for b in B:
        for c in C:
            res.append([a,b,c])

如果我有4个这样的列表，这样可以嵌套4个for

但如果我列表数是不固定的呢，如果有100个列表，我不能手动把for循环嵌套100次吧，有什么好方法呢？

Answer 1

from itertools import product
loop_val = [a,b,c,d]
rst = list(product(*loop_val))