1. ホーム
  2. python

[解決済み] 繰り返しのある順列の生成

2022-10-20 16:19:35

質問

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)