111

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

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

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

4

6 に答える 6

181

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

数学では、デカルト積 (または積集合) は 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)
于 2010-06-23T08:20:54.727 に答える
33

順列を探しているのではなく、Cartesian Productが必要です。これにはitertools の製品を使用します。

from itertools import product
for roll in product([1, 2, 3, 4, 5, 6], repeat = 2):
    print(roll)
于 2010-06-23T08:21:43.243 に答える
11

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)]
于 2010-06-23T09:27:48.550 に答える