Python生成排列组合

生成排列可以用product:

1
2
3
4
from itertools import product
l = [1, 2, 3]
print list(product(l, l))
print list(product(l, repeat=4))

组合的话可以用combinations:

1
2
from itertools import combinations
print list(combinations([1,2,3,4,5], 3))

下面是我以为没有combinations然后自己写的,没有itertools的python(2.6以下)可供参考。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import copy

def 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 answers

print combine([1, 2, 3, 4, 5], 3)

输出:

1
[[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]]
------ 本文结束------
坚持原创技术分享,您的支持将鼓励我继续创作!

欢迎关注我的其它发布渠道