python列表实现元素的排列组合

# 现有一个业务场景，需要对序列中的元素两两组合
# 自己实现了之后，发现有现成的方法可以调用。
# 以下将自己写的与网上百度到的都放到这里。

# 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)