繰り返しによる順列の生成


84

itertoolsについては知っていますが、繰り返しなしでしか順列を生成できないようです。

たとえば、2つのサイコロで可能なすべてのサイコロを生成したいとします。したがって、繰り返しを含む[1、2、3、4、5、6]のサイズ2のすべての順列が必要です:(1、1)、(1、2)、(2、1)...など

可能であれば、これを最初から実装したくありません

回答:


144

デカルト積を探しています。

数学では、デカルト積(または積セット)は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)

8
これは、2つのサイコロを振る非常に非効率的な方法です…2回の呼び出しrandom.randintは、より簡単で効率的です。
Eric O Lebigot 2010年

可能なすべてのペアを生成しない場合、ランダムなサイコロの目ははるかに速くなります:[random.randint(1,6)for i in xrange(2)]
liori 2010年

13
私は実際にランダムなロールを生成しようとしていませんでした。すべての可能なロールをリストするためだけです。
bwmat 2010年


7

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

12
このソリューションは、組み合わせに出失い(2, 1)(3, 2)(3, 1)および類似...一般的には、第2のロールは、最初のより低いすべての組み合わせを残します。
holroy 2015年

1

この場合、リスト内包表記は特に必要ありません。

与えられた

import itertools as it


seq = range(1, 7)
r = 2

コード

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

詳細

明らかに、デカルト積は順列のサブセットを生成できます。ただし、次のようになります。

  • 交換に:すべての順列N生み出す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


-1

私は私だけ使用して解決策を見つけたと思うlambdasmapreduce

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

-2

まず、itertools.permutations(list)によって返されるジェネレーターを最初にリストに変換する必要があります。次に、set()を使用して、重複を削除できます。次のようになります。

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

1
重複は含まれません。
ビョルンLindqvist

1
OPは明示的に複製を望んでいます
Levi Lesches 2018年
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.