在python列表理解中枚举三个变量

Question

我正在尝试打印三个变量的列表的所有可能枚举。例如，如果我的输入是：

x = 1
y = 1
z = 1

我希望输出是这样的：

[[0, 0, 0], [0, 0, 1], [0, 1, 0], [1, 0, 0], [1, 1, 0], [1, 0, 1], [0, 1, 1], [1, 1, 1]]

如果 x,y,z 变量中的任何一个大于 1，它将枚举从 0 到变量值的所有整数。例如，如果 x=3，则 0、1、2 或 3 可能位于 3 元素列表的第一个槽中。

现在我正在创建这样的列表理解：

output = [ [x,y,z] for x,y,z in range(x,y,z)]

我认为 range 函数有问题？

Answer 1

您可以使用itertools 中的product() 函数，如下所示：

from itertools import product

answer = list(list(x) for x in product([0, 1], repeat=3))
print(answer)

输出

[[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1], [1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 1]]

Answer 2

作为使用 product 的解决方案的补充，您还可以使用三重列表推导式。

>>> x, y, z = 1, 2, 3
>>> [(a, b, c) for a in range(x+1) for b in range(y+1) for c in range(z+1)]
[(0, 0, 0),
 (0, 0, 1),
 (0, 0, 2),
 (some more),
 (1, 2, 2),
 (1, 2, 3)]

+1 是必需的，因为range 不包括上限。 如果您希望输出为列表列表，您可以使用[[a, b, c] for ...]。

但是请注意，这显然只有在您始终拥有三个变量（x、y、z）时才有效，而product 将适用于任意数量的列表/上限。

Answer 3

您可以在列表理解中使用range() 函数和itertools.product 函数：

>>> x = 1
>>> y = 1
>>> z = 1
>>> from itertools import product
>>> list(product(*[range(i+1) for i in [x,y,z]]))
[(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)]

这种方法也适用于不同的数字：

>>> x = 2
>>> y = 2
>>> z = 2
>>> 
>>> list(product(*[range(i+1) for i in [x,y,z]]))
[(0, 0, 0), (0, 0, 1), (0, 0, 2), (0, 1, 0), (0, 1, 1), (0, 1, 2), (0, 2, 0), (0, 2, 1), (0, 2, 2), (1, 0, 0), (1, 0, 1), (1, 0, 2), (1, 1, 0), (1, 1, 1), (1, 1, 2), (1, 2, 0), (1, 2, 1), (1, 2, 2), (2, 0, 0), (2, 0, 1), (2, 0, 2), (2, 1, 0), (2, 1, 1), (2, 1, 2), (2, 2, 0), (2, 2, 1), (2, 2, 2)]
>>> 

Answer 4

如果您需要以列表列表（而不是元组列表）的形式使用它，您可以在 Kasramvd 的答案输出上使用 map，即：

map(list,list(product(*[range(i+1) for i in [x,y,z]])))