星期二, 4月 17, 2012

求出n取k組合的列表 Python 版

這版本寫完,我就要認真的開始寫我的作業。所以這篇草草帶過:
一般的組合:
from itertools import combinations
all=3
want=2
for values in combinations(range(1, all+1), want):
    print (values)
結果:
(1, 2)
(1, 3)
(2, 3)
特別格式的組合:
from itertools import permutations
all=3
want=2
list=[]
for i in range(want):
    list.append(1)
for i in range(all-want):
    list.append(0)
for values in set(permutations(list)):
    print (values)
結果:
(0, 1, 1)
(1, 1, 0)
(1, 0, 1)

沒有留言: