Перечисление трех переменных в понимании списка 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]] 

Если какой-либо из х, у, г переменных больше 1, было бы перечислить все целые числа от 0 до значения переменной. Например, если x = 3, то 0, 1, 2 или 3 будут возможны в первом слоте списков из 3 элементов.

Сейчас я создаю список понимание, как это:

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

я думаю, что-то не так с функцией диапазона?

Answer 1

В дополнение к решениям с помощью 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 2

Вы можете использовать функцию product() из itertools следующим образом:

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 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

Если вам это нужно в виде списка списков (вместо списка кортежей), вы можете использовать map над выводом ответа по Kasramvd, то есть:

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