# 现有一个业务场景,需要对序列中的元素两两组合
# 自己实现了之后,发现有现成的方法可以调用。
# 以下将自己写的与网上百度到的都放到这里。
# PART1 自己写的
# 思路:每次弹出(pop)序列中的一个元素,让其依次(遍历)与余下的元素组合
def comb(a, seq):
"""
:param a: init中每次弹出的元素
:param seq: init中每次弹出的元素之后的序列
:return:
"""
return [(a, i) for i in seq]
def start(init):
li = []
for i in range(len(init) - 1): # 必须得减1,因为是pop,而最后一次传入到combine中时的序列必须不为空序列
t = comb(init.pop(), init)
li.extend(t)
return li
test = ['a', 'b', 'c']
print(start(test))
# PART2 百度到的
import itertools
import itsdangerous # 导上面的包时发现的,不知道什么用,先放着这里
aa = ['a', 'b', 'c', 2020]
amount = 3 # 每个组合的元素个数,比如设置为2就是两两组合
bb = list(itertools.permutations(aa, amount)) # 排列
print(bb)
print("######################")
cc = list(itertools.combinations(aa, amount)) # 组合
print(cc)
python列表实现元素的排列组合
最新推荐文章于 2024-06-27 23:55:41 发布