itertoolsについては知っていますが、繰り返しなしでしか順列を生成できないようです。
たとえば、2つのサイコロで可能なすべてのサイコロを生成したいとします。したがって、繰り返しを含む[1、2、3、4、5、6]のサイズ2のすべての順列が必要です:(1、1)、(1、2)、(2、1)...など
可能であれば、これを最初から実装したくありません
回答:
デカルト積を探しています。
数学では、デカルト積(または積セット)は2つのセットの直接積です。
あなたの場合、これは{1, 2, 3, 4, 5, 6}
xになり{1, 2, 3, 4, 5, 6}
ます。
itertools
そこであなたを助けることができます:
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)]
ランダムなサイコロを振るには(完全に非効率的な方法で):
import random
random.choice([p for p in itertools.product(x, repeat=2)])
(6, 3)
Python 2.7および3.1には、次のitertools.combinations_with_replacement
関数があります。
>>> 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)]
(2, 1)
、(3, 2)
、(3, 1)
および類似...一般的には、第2のロールは、最初のより低いすべての組み合わせを残します。
この場合、リスト内包表記は特に必要ありません。
与えられた
import itertools as it
seq = range(1, 7)
r = 2
コード
list(it.product(seq, repeat=r))
詳細
明らかに、デカルト積は順列のサブセットを生成できます。ただし、次のようになります。
product
置換を伴う順列、n r
[x for x in it.product(seq, repeat=r)]
置換なしの順列、n!
[x for x in it.product(seq, repeat=r) if len(set(x)) == r]
# Equivalent
list(it.permutations(seq, r))
したがって、すべての組み合わせ関数は次の場所から実装できますproduct
。
combinations_with_replacement
から実装 product
combinations
から実装されpermutations
、で実装できますproduct
(上記を参照)私は私だけ使用して解決策を見つけたと思うlambdas
、map
と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)), [])
基本的に、行を指定して列を反復する最初のラムダ関数をマッピングしています
list(map(lambda j: (i, j), np.arange(n)))
次に、これは新しいラムダ関数の出力として使用されます
lambda i:list(map(lambda j: (i, j), np.arange(n)))
可能なすべての行にマップされます
map(lambda i: list(map(lambda j: (i, j), np.arange(n))), np.arange(m))
次に、結果のすべてのリストを1つに減らします。
2つの異なる番号を使用することもできます。
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)), [])
まず、itertools.permutations(list)によって返されるジェネレーターを最初にリストに変換する必要があります。次に、set()を使用して、重複を削除できます。次のようになります。
def permutate(a_list):
import itertools
return set(list(itertools.permutations(a_list)))
random.randint
は、より簡単で効率的です。