pythonを用いた、組み合わせ列挙の方法

Question

pythonを用いて、
組み合わせの一覧を表示するプログラムを
作りたいと考えております。

例えば、

a=[1, 2, 3]
b=[5, 6]

という２つの配列に対して、

[[1, 5], [1, 6], [2, 5], [2, 6], [3, 5], [3, 6]]

のような結果を返すものです。

この例のように、
予め、配列の個数がわかっていれば、
for文を用いて

data = []
for i in a:
    for j in b:
        data.append([i, j])

とするか、
もしくはitertoolsを使って、

data = list(itertools.product(a, b))

とすることができます。

しかし、今回厄介なのが、
この配列の個数が状況によって変わる
(データをファイルから読み込むようなものを考えているので、
読み込むファイルで変わる)
というこであり、
場合によっては、

a=[1, 2, 3]
b=[5, 6]
c=[9, 8, 7, 6, 5]

であったり

a=[1, 2, 3]
b=[5, 6]
c=[9, 8, 7, 6, 5]
d=[11, 12]
e=[1, 2, 3, 4, 5]

だったりするわけです。
実際これらの配列は、
辞書に格納しようと考えており、

dic={'a':[1, 2, 3], 'b':[5, 6]}

このような状態から

[[1, 5], [1, 6], [2, 5], [2, 6], [3, 5], [3, 6]]

を得たいと考えております。
始めは、単に入れ子構造にすれば良いと考え、

dic={'a':[1, 2, 3], 
     'b':[5, 6], 
     'c':[9, 8, 7, 6, 5], 
     'd':[11, 12]}
dick=dic.keys()
data = itertools.product(dic[dick[0]], dic[dick[1]]) 
for dk in range(len(dick)-2):
    data = itertools.product(data, dic[dick[dk+2]]) 

のようにしようと考えました。
しかし、この場合、
次のようになってしまいます。

[(((1, 9), 5), 11),
 (((1, 9), 5), 12),
 (((1, 9), 6), 11),
 ...

一方イメージしているのは
次のような感じです。

[(1, 9, 5, 11),
 (1, 9, 5, 12),
 (1, 9, 6, 11),
...

以上長くなってしまいましたが、
みなさんのお力をお借りしたいです。
どうかよろしくお願いいたします。

Answer 1

可変長引数に対して配列を引数展開すればいいと思います。

data = [[1, 2, 3], [5, 6], [9, 8, 7, 6, 5]]
list(itertools.product(*data))
#=> [(1, 5, 9), (1, 5, 8), (1, 5, 7), (1, 5, 6), (1, 5, 5), (1, 6, 9), (1, 6, 8), (1, 6, 7), (1, 6, 6), (1, 6, 5), (2, 5, 9), (2, 5, 8), (2, 5, 7), (2, 5, 6), (2, 5, 5), (2, 6, 9), (2, 6, 8), (2, 6, 7), (2, 6, 6), (2, 6, 5), (3, 5, 9), (3, 5, 8), (3, 5, 7), (3, 5, 6), (3, 5, 5), (3, 6, 9), (3, 6, 8), (3, 6, 7), (3, 6, 6), (3, 6, 5)]

Comments

[A5size]

恥ずかしながら、
可変長引数に対して
こんな引数の渡し方があることを知りませんでした。
まさにやりたかったことです。
非常に助かりました。
ありがとうございます。

Answer 2

dic={'a':[1, 2, 3], 
     'b':[5, 6], 
     'c':[9, 8, 7, 6, 5], 
     'd':[11, 12]}
oldresult=[[]]
for k in dic:
    result=[]
    for r in oldresult:
        for i in dic(k):
            result.append(r+[i])
    oldresult=result
print result