python怎么生成list的所有元素的組合比如l = [1,2,3]指定長(zhǎng)度2，就會(huì)得到[1,2][2,3][1,3]，不考慮順序。
2 回答
絕地?zé)o雙
TA貢獻(xiàn)1946條經(jīng)驗(yàn) 獲得超4個(gè)贊
生成排列可以用product：
from itertools import productl = [1, 2, 3]print list(product(l, l))print list(product(l, repeat=4))
組合的話可以用combinations：
from itertools import combinationsprint list(combinations([1,2,3,4,5], 3))
下面是我以為沒(méi)有combinations然后自己寫的，沒(méi)有itertools的python（2.6以下）可供參考。
import copydef combine(l, n): answers = [] one = [0] * n def next_c(li = 0, ni = 0): if ni == n: answers.append(copy.copy(one)) return for lj in xrange(li, len(l)): one[ni] = l[lj] next_c(lj + 1, ni + 1) next_c() return answersprint combine([1, 2, 3, 4, 5], 3)
輸出：
[[1, 2, 3], [1, 2, 4], [1, 2, 5], [1, 3, 4], [1, 3, 5], [1, 4, 5], [2, 3, 4], [2, 3, 5], [2, 4, 5], [3, 4, 5]]
慕斯王
TA貢獻(xiàn)1864條經(jīng)驗(yàn) 獲得超2個(gè)贊
import itertools nums = itertools.permutations([1,2,3])for x in nums: print(x)>>> (1, 2, 3) (1, 3, 2) (2, 1, 3) (2, 3, 1) (3, 1, 2) (3, 2, 1)
添加回答
舉報(bào)
0/150
提交
取消