python3 組み合わせについて(重複なし)

Question

お世話になっております。
重複なしの組み合わせのプログラムを作成しております。
下記のようなことを実装できるモジュールを作成したいのですが、
プログラム自体初学者故考えが及ばないので、質問させてもらいます。

['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i']

上記リストを[2,3,4]といった値を与えられたときに
9C2 * 7C3 * 4C4といった重複なしの組み合わせのリストを作成できるモジュールを作りたいです。

例)
[[['a', 'b'], ['c', 'd', 'e'], ['f', 'g', 'h', 'i']], [['a', 'b'], ['c', 'd', 'f'], ['e', 'g', 'h', 'i']], [['a', 'b'], ['c', 'd', 'g'], ['e', 'f', 'h', 'i']], ...

def comb(a,b) #aに[2,3,4]や[5,4],[6,3]等が入る想定。bには文字列のリスト

combinationsを使用して、9C2,7C3,4C4それぞれの組み合わせを表示するだけなら出来ましたが、
そこから上記例のようなリストにまとめることができません。

ぜひご回答頂ければと思います。よろしくお願いいたします。

Answer 1

再帰で書くと比較的簡潔に書けるかと思います。

import itertools

# a  : 各組合せ数のリスト
# b  : 組み合わせる要素リスト
# l  : 途中までの結果リスト
# lv : 現在の組み合わせ数の位置
#
def comb(a, b, l, lv):
    if not b:
        print(l) # 最終結果をリストとして保持したいなら、このlをグローバルに用意したリストなりに追加すればよい
        return

    for ret in itertools.combinations(b, a[lv]):
        lc = l.copy()
        lc.append(ret)
        comb(a, b - set(ret), lc, lv + 1)

a = [2, 3, 4]
b = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i']

#a = [2, 3]
#b = ['a', 'b', 'c', 'd', 'e']
comb(a, set(b), [], 0)

Comments

[Sayre]

回答ありがとうございます。
実際に欲しい値が出力されることを確認しました。
また一つ教えていただきたいのですが、もしcombの関数の引数が
二つしか取れない場合(aとbのみ)はどうすればよろしいでしょうか？

[can110]

なぜそのような制約がある（必要？）のでしょうか？

[Sayre]

完全に個人的な話になってしまい申し訳ありませんが、
プライベートで取り組んでいるpythonの問題にそのような制約があるためです。。

[can110]

課題的な問題でしたらヒントだけ示します。
グローバルな変数を使うのもひとつの手です。
あまりエレガントではないですが。。

[Sayre]

ありがとうございます。
グローバルな変数を使ってみます。

Answer 2

全体の順列を取って、スライスで分割するのが楽だと思います。

import itertools


src = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i']

for b in itertools.permutations(src):
    b = b[:2], b[2:5], b[5:]
    print(b)

実行結果(抜粋) Wandbox

(('a', 'b'), ('c', 'd', 'e'), ('f', 'g', 'h', 'i'))
(('a', 'b'), ('c', 'd', 'e'), ('f', 'g', 'i', 'h'))
(('a', 'b'), ('c', 'd', 'e'), ('f', 'h', 'g', 'i'))
...
(('a', 'c'), ('b', 'd', 'e'), ('f', 'g', 'h', 'i'))
(('a', 'c'), ('b', 'd', 'e'), ('f', 'g', 'i', 'h'))
(('a', 'c'), ('b', 'd', 'e'), ('f', 'h', 'g', 'i'))
...

ただしこの場合処理が 9! = 362880 回繰り返されるので、現実的であるかは微妙です。

Comments

[Sayre]

解答ありがとうございます。
今回は重複はなしで考えたいので順列ではない想定です。
なので、今回[2,3,4]の組み合わせを考える場合、
9C2 * 9C3* 9C4 = 1260通りになる想定です。

[LouiS0616]

ああ、内側のリストの要素は順を問わないのですね。
それならば確かに 9! / (2! x 3! x 4!) で 1260通りです。