如何在python中使用列表/数组理解来生成交叉产品对

Question

是否可以使用列表理解或数组理解来生成数组数组？

例如，我有一个从 0 到 9 的列表：

rng = range(0,10)

然后我想从上面的 rng 创建一个 5 对 cartesion 产品对。我尝试了以下但它不起作用。

[arr for (for i in range(0,5) arr[i] in rng)]   

它不编译。有人可以让我知道编码的正确方法吗？

Answer 1

要使用列表理解来做到这一点：

[ (x, y) for x in rangeA for y in rangeB ]

itertools.product不过，我会使用，因为它更能自我记录。

重新阅读您的问题后，您似乎还想计算任意维度的叉积，这是无法在列表理解中参数化的。