Tôi biết về itertools, nhưng có vẻ như nó chỉ có thể tạo ra các hoán vị mà không lặp lại.

Ví dụ: tôi muốn tạo tất cả các cuộn xúc xắc có thể có cho 2 viên xúc xắc. Vì vậy, tôi cần tất cả các hoán vị có kích thước 2 của [1, 2, 3, 4, 5, 6] bao gồm các lần lặp lại: (1, 1), (1, 2), (2, 1) ... vv

Nếu có thể, tôi không muốn thực hiện điều này từ đầu

Bạn đang tìm kiếm Sản phẩm Descartes .

Trong toán học, tích Descartes (hay tập hợp sản phẩm) là tích trực tiếp của hai tập hợp.

Trong trường hợp của bạn, đây sẽ là {1, 2, 3, 4, 5, 6}x {1, 2, 3, 4, 5, 6}. itertoolscó thể giúp bạn ở đó:

import itertools
x = [1, 2, 3, 4, 5, 6]
[p for p in itertools.product(x, repeat=2)]
[(1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6), (2, 1), (2, 2), (2, 3), 
 (2, 4), (2, 5), (2, 6), (3, 1), (3, 2), (3, 3), (3, 4), (3, 5), (3, 6), 
 (4, 1), (4, 2), (4, 3), (4, 4), (4, 5), (4, 6), (5, 1), (5, 2), (5, 3), 
 (5, 4), (5, 5), (5, 6), (6, 1), (6, 2), (6, 3), (6, 4), (6, 5), (6, 6)]

Để nhận được một cuộn xúc xắc ngẫu nhiên (theo một cách hoàn toàn không hiệu quả ):

import random
random.choice([p for p in itertools.product(x, repeat=2)])
(6, 3)

Bạn không tìm kiếm hoán vị - bạn muốn Sản phẩm Descartes . Đối với sản phẩm sử dụng này từ itertools:

from itertools import product
for roll in product([1, 2, 3, 4, 5, 6], repeat = 2):
    print(roll)

Trong python 2.7 và 3.1 có một itertools.combinations_with_replacementchức năng:

>>> list(itertools.combinations_with_replacement([1, 2, 3, 4, 5, 6], 2))
[(1, 1), (1, 2), (1, 3), (1, 4), (1, 5), (1, 6), (2, 2), (2, 3), (2, 4), 
 (2, 5), (2, 6), (3, 3), (3, 4), (3, 5), (3, 6), (4, 4), (4, 5), (4, 6),
 (5, 5), (5, 6), (6, 6)]

Trong trường hợp này, việc hiểu danh sách là không đặc biệt cần thiết.

Được

import itertools as it


seq = range(1, 7)
r = 2

Mã

list(it.product(seq, repeat=r))

Chi tiết

Rõ ràng, tích Descartes có thể tạo ra các tập con các hoán vị. Tuy nhiên, nó theo sau rằng:

• với sự thay thế: tạo ra tất cả các hoán vị n ^r quaproduct
• không cần thay thế: bộ lọc từ cái sau

Hoán vị có thay thế, n ^r

[x for x in it.product(seq, repeat=r)]

Hoán vị không thay thế, n!

[x for x in it.product(seq, repeat=r) if len(set(x)) == r]

# Equivalent
list(it.permutations(seq, r))  

Do đó, tất cả các hàm tổ hợp có thể được thực hiện từ product:

• combinations_with_replacement thực hiện từ product
• combinationsđược triển khai từ permutations, có thể được triển khai với product(xem ở trên)

Tôi nghĩ rằng tôi đã tìm thấy một giải pháp chỉ sử dụng lambdas, mapvà reduce.

product_function = lambda n: reduce(lambda x, y: x+y, map(lambda i: list(map(lambda j: (i, j), np.arange(n))), np.arange(n)), [])

Về cơ bản, tôi đang ánh xạ một hàm lambda đầu tiên cho một hàng, lặp lại các cột

list(map(lambda j: (i, j), np.arange(n)))

thì nó được sử dụng làm đầu ra của một hàm lambda mới

lambda i:list(map(lambda j: (i, j), np.arange(n)))

được ánh xạ trên tất cả các hàng có thể có

map(lambda i: list(map(lambda j: (i, j), np.arange(n))), np.arange(m))

và sau đó chúng tôi giảm tất cả các danh sách kết quả thành một.

thậm chí còn tốt hơn

Cũng có thể sử dụng hai số khác nhau.

prod= lambda n, m: reduce(lambda x, y: x+y, map(lambda i: list(map(lambda j: (i, j), np.arange(m))), np.arange(n)), [])

Trước tiên, bạn sẽ muốn biến trình tạo được trả về bởi itertools.permutations (danh sách) thành một danh sách trước. Sau đó, thứ hai, bạn có thể sử dụng set () để xóa các bản sao giống như bên dưới:

def permutate(a_list):
    import itertools
    return set(list(itertools.permutations(a_list)))