Python itertools模块combinations方法

  1、Python itertools模块combinations(iterable, r)方法可以创建一个迭代器,返回iterable中所有长度为r的子序列,返回的子序列中的项按输入iterable中的顺序排序。

import itertools
list1 = [1, 3, 4, 5]
list2 = list(itertools.combinations(list1, 2))
print(list2)

返回结果:
[(1, 3), (1, 4), (1, 5), (3, 4), (3, 5), (4, 5)]

  2、实现一组数据的所有排列组合

import itertools

list1 = [1, 3, 4, 5]
list2 = []
for i in range(1, len(list1)+1):
    iter1 = itertools.combinations(list1, i)
    list2.append(list(iter1))
print(list2)


返回结果:
[[(1,), (3,), (4,), (5,)], [(1, 3), (1, 4), (1, 5), (3, 4), (3, 5), (4, 5)], [(1, 3, 4), (1, 3, 5), (1, 4, 5), (3, 4, 5)], [(1, 3, 4, 5)]]

 

  

posted @ 2018-11-06 16:27  FENG4Y  阅读(22611)  评论(1编辑  收藏  举报