美文网首页
python 内置排序函数使用

python 内置排序函数使用

作者: random_walk | 来源:发表于2021-05-23 15:58 被阅读0次

    python内置关于排序的工具主要有两个一个是列表自带的sort()方法,另外一个是sorted()函数。Python 列表内置方法可以直接修改列表。而sorted()内置函数从一个可迭代对象(列表,元组等都可以)构建一个新的排序列表。其函数原型分别如下:

    sort(*, key=None, reverse=False)
    sorted(iterable, *, key=None, reverse=False)
    

    简单使用

    对列表进行默认排序

    a = [1, 4, 2, 5, 3]
    a.sort() # 默认升序排列
    a.sort(reverse=True) # 降序排列 
    b = sorted(a)
    

    自定义排序方法

    从函数原型来看,可以看到两者都具有两个可选参数,它们都必须指定为关键字参数。
    key指定带有单个参数的函数,用于从 iterable 的每个元素中提取用于比较的键 (例如 key=str.lower)。默认值为 None (直接比较元素)。key形参的值应该是个函数(或其他可调用对象),它接受一个参数并返回一个用于排序的键。

    用lambda函数实现自定义排序

    假设有其他类型的变量,比如一个自定义的类或者列表中又是一个列表。以官网例子为例有这样一个列表,其元素为元组,

    student_tuples = [
        ('john', 'A', 15),
        ('jane', 'B', 12),
        ('dave', 'B', 10),
    ]
    

    可以用以下方式按照年龄排序

    sorted(student_tuples, key=lambda student: (firstkey, secondkey) )
    sorted(student_tuples, key=lambda student: student[2])
    sorted(student_tuples, key=lambda student: (student[2], student[1])) //定义优先级
    

    类似的有自定义类

    class Student:
        def __init__(self, name, grade, age):
            self.name = name
            self.grade = grade
            self.age = age
        def __repr__(self):
            return repr((self.name, self.grade, self.age))
    

    可以用如下方式进行排序

    sorted(student_objects, key=lambda student: student.age)
    

    也可以显示定义一个函数,且只有一个参数,返回用于排序的键,比如

    student_tuples = [
        ('john', 'A', 15),
        ('jane', 'B', 12),
        ('dave', 'B', 10),
    ]
    
    def cmp(x):
      return x[2]
      
    student_tuples.sort(key = cmp)
    

    总之就是定义一个函数返回一个用于排序的键,可以用lambda函数或者def定义都可以。

    使用operator模块函数

    上面实现的简单函数实际就是实现了返回一个有序结构的第n的元素,或者某个类中的某个属性,因此 Python 提供了便利功能,使访问器功能更容易,更快捷。operator 模块有 itemgetter() 、 attrgetter() 函数。分别完成返回第n个元素,某个属性功能。上面的排序可以用如下方式进行实现

    from operator import itemgetter, attrgetter
    sorted(student_tuples, key=itemgetter(2))
    sorted(student_objects, key=attrgetter('age'))
    

    老式自定义排序方法

    在python2中,sort有一个cmp参数,即用一个函数来自定义比较,在python3中这种方式被取消。为了继承类似的用法,在 Python 3.2 中, functools.cmp_to_key() 函数被添加到标准库中的 functools模块中。
    这种作用先定义如何比较两个变量,以上面的学生列表按照年龄排序为例

    from functools import cmp_to_key
    
    student_tuples = [
        ('john', 'A', 15),
        ('jane', 'B', 12),
        ('dave', 'B', 10),
    ]
    
    def cmp(x, y):
      return x[2] - y[2]
      
    student_tuples.sort(key = cmp_to_key(cmp))
    

    这种做法自定义比较函数接收两个形参,返回比较结果(bool),而新式方法接受一个参数,返回的是比较的键。

    对字典进行排序

    假设有字典d = {'b':2, 'a':1,'c':8,'d':4},则可以通过以下方式对字典按照键和值进行排序

    dict(sorted(x.items(), key=lambda item: item[0]))  // 按照key
    dict(sorted(x.items(), key=lambda item: item[1])) //按照value
    或
    {k: v for k, v in sorted(x.items(), key=lambda item: item[0])}
    {k: v for k, v in sorted(x.items(), key=lambda item: item[1])}
    

    key传入类

    class Key:
        def __init__(self,a):
            self.a = a
        def __lt__(self, other):
            return self.a+other.a < other.a+self.a
    

    相关文章

      网友评论

          本文标题:python 内置排序函数使用

          本文链接:https://www.haomeiwen.com/subject/jdqzjltx.html