python - 来自 n 个数组的组合从每个数组中选取一个元素

Question

虽然我在下面的链接中找到了答案，但我想知道如何在 python 中使用 itertools 来实现这一点。通过从每个数组中选择一个元素来使用 n 个数组的组合

标签： pythoncombinationsitertools

Answer 1

import itertools    

A = [[1], [2,3,4], [5]]
result = []
for c in itertools.product(*A):
    result.append(list(c))
print(result)

参考：https ://docs.python.org/3/library/itertools.html#itertools.product