如何改进此列表理解？答案

【问题标题】：How to improve this List Comprehensions?如何改进此列表理解？
【发布时间】：2020-04-25 14:16:42
【问题描述】：

我想知道我可以改进我的 python 代码以使其更短。我只使用标准库和 numpy 库。这是代码。

import numpy as np
x=1
y=2
np.array([(a,b,c) for a in [x,y] for b in [x,y] for c in [x,y]])

如您所见，该程序生成了所有三种元素组合。这个程序中有类似的部分，如for a in [x, y]，可以写得更短吗？

[已编辑]

我只想使用标准库和 numpy 库。

输入应该是任何值，例如 x=3、y=8。

【问题讨论】：

这能回答你的问题吗？ Numpy: efficient way to generate combinations from given ranges
Numpy: efficient way to generate combinations from given ranges 在生成所有值的组合方面类似。但它们的输入值是连续数。我想使用任何值作为输入。（我忘了在我的问题中写下这个条件，抱歉。）
基于Junkrat 建议的类似问题，我提出了这个想法。我写了代码作为答案之一。

【解决方案1】：

使用itertools.product:

from itertools import product

# ...
np.array(list(product((x, y), repeat=3)))

【讨论】：

【解决方案2】：

基于Junkrat 建议的类似问题，我提出了这个想法。（但这并不简单。）

import numpy as np
x=1
y=2
a = np.array(np.meshgrid([x, y], [x, y], [x, y]))   
a = np.rollaxis(a, 0, 4)         
a = a.reshape((2*2*2, 3))

【讨论】：