Python3+itertools实现排列组合教程

一、说明

本文的直接起因是上周公司的一个比赛用到了排列组合,之前没用过,这里记一记。

本文说的排列组合是借助itertools实现,而不是自己写代码实现。

itertools的其他一些函数还是比较有意思的,所以在最后也会做下简单的介绍。

 

二、排列组合实现

功能需求 总数算法 实现函数 示例 示例输出
元素可重复使用的排列 n的m次方 product(*iterables, repeat=1) product('ABCD', repeat=2) AA AB AC AD BA BB BC BD CA CB CC CD DA DB DC DD
元素不可重复使用的排列 A n m permutations(iterables, r=None) permutations('ABCD', 2) AB AC AD BA BC BD CA CB CD DA DB DC
元素可重复使用的组合 C n m + n combinations_with_replacement(iterable, r) combinations_with_replacement('ABCD', 2) AA AB AC AD BB BC BD CC CD DD
元素不可重复使用的组合 C n m combinations(iterable, r) combinations('ABCD', 2) AB AC AD BC BD CD
 

说明:

1. 上边函数的第一个参数iterables表示可迭代的对象,包括字符串、列表等等。所以排列组合的元素如果不是字母而是字符串也是可以的,如permutations(['AB', 'CD', 'EFG'], 2)。

2. 上边函数的第二个参数都是参与排列组合的元素个数。

 

三、itertools其他函数

3.1 无穷类函数

功能 实现函数 示例 示例输出
从某个数开始,按给定步长一直数下去 count(start=0, step=1) count(10, 2) 10 12 14 ...
不断循环输出给出的元素 cycle(iterable) cycle('ABC') A B C A B C ...
不断重复/重复给定次数给定元素 repeat(object[, times]) repeat(10, 3) 10 10 10
 

3.2 分组遍历函数---groupby()

假设我们有一个如下字典组成的列表,列表中每个字典记录了每个学生的一门课程成绩,在遍历时我们想先打印A同学的各科成绩,再打印B同学的各科成绩。这时就可借助groupby()函数实现这个需求。

from itertools import groupby

student_score_dict_list = [
    {
        "student_name": "A",
        "course_name": "语文",
        "scores": 80
    },
    {
        "student_name": "B",
        "course_name": "数学",
        "scores": 91
    },
    {
        "student_name": "B",
        "course_name": "语文",
        "scores": 90
    },
    {
        "student_name": "A",
        "course_name": "数学",
        "scores": 81
    },
]

# 定义一个函数,这个函数的作用是传给他一个值,它返回代表这个值的值或表表参与比较等操作
# 如果是列表那就先按第一个排,第一个值相同再按第二个值排
# 用lambda表达示还是用def效果一样的
# 排序使用的选取代表值的方法,我们先以学生名排序,再以课程名排序
# 这里有点小尴尬的是在编码上“数学”比“语文”小,所以“数学”会排前面;如果是数值可以在前面加个负号,字符串还不好处理
sort_key_func = lambda x: (x["student_name"], x["course_name"])
# 分组时使用的选取代表值的方法,我们先以学生名进行分组
groupby_key_func = lambda x: x["student_name"]

sorted_student_score_dict_list = sorted(student_score_dict_list, key=sort_key_func)
groupby_student_score_dict_list = groupby(sorted_student_score_dict_list, key=groupby_key_func)

for student, one_student_score_dict_list in groupby_student_score_dict_list:
    print(f"{student} scores: ")
    for student_score_dict in one_student_score_dict_list:
        print(f"\t{student_score_dict['course_name']}: {student_score_dict['scores']}")
View Code

实现效果如下:

 

 

参考:

https://docs.python.org/3/library/itertools.html#module-itertools

posted on 2021-04-30 17:44  诸子流  阅读(945)  评论(0编辑  收藏  举报