Liệt kê ba biến trong việc hiểu danh sách python

Question

Tôi đang cố gắng in tất cả các liệt kê có thể có của một danh sách cho ba biến. Ví dụ, nếu đầu vào của tôi là:

x = 1 
y = 1 
z = 1 

Tôi muốn đầu ra được như:

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

Nếu bất kỳ x, y, biến z là cao hơn 1, nó sẽ liệt kê tất cả các số nguyên từ 0 đến giá trị biến. Ví dụ: nếu x = 3 thì 0, 1, 2 hoặc 3 sẽ có thể ở vị trí đầu tiên trong danh sách 3 phần tử.

Ngay bây giờ tôi đang tạo ra danh sách hiểu như thế này:

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

Tôi nghĩ rằng cái gì là sai với chức năng nhiều?

Answer 1

bổ sung các giải pháp sử dụng product, bạn cũng có thể sử dụng danh sách hiểu ba.

>>> 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 là cần thiết vì range không bao gồm giới hạn trên. Nếu bạn muốn đầu ra là danh sách các danh sách, bạn chỉ có thể thực hiện [[a, b, c] for ...].

Lưu ý, tuy nhiên, điều này làm việc sẽ rõ ràng duy nhất là bạn luôn có ba biến (x, y, z), trong khi product sẽ làm việc với một số tùy ý các danh sách/giới hạn trên.

Answer 2

Bạn có thể sử dụng product() chức năng từ itertools như sau:

from itertools import product 

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

Output

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

Bạn có thể sử dụng range() chức năng trong một sự hiểu biết danh sách và itertools.product chức năng:

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

Cách tiếp cận này sẽ làm việc cho các số khác nhau quá:

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

Nếu bạn cần nó trong hình thức của một danh sách liệt kê (thay vì danh sách các hàng), bạn có thể sử dụng map so với sản lượng của câu trả lời bằng cách Kasramvd, đó là:

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