python - 使用标准库在运行时创建 ND 数组

Question

我有三个数组

a = [2]
b = [2,3,6]
c = [1]

我想合并它们，以便得到一个大小为 len(a)*len(b) 的数组，其中包含两者的所有排列。 (C 将始终包含单个值)

我认为这样的事情会奏效

newArr = [for i in range len(a)*len(b) [for x in a][for y in b][for z in c]]
print(newArr)

[[2,2,1],[2,3,1],[2,6,1]]

但是它似乎不允许在语言的语法中使用它。有人知道我如何使用标准库执行此操作吗？

Answer 1

[[x, y, z] for x in a for y in b for z in c]

例如:

>>> [[x, y, z] for x in [2] for y in [2,3,6] for z in [1]]
[[2, 2, 1], [2, 3, 1], [2, 6, 1]]