首页 > 技术文章 > python 对列表里面的对象,类元素排序

hq0803 2021-12-20 10:37 原文

下边是一个例子,对Departs列表中的元素(均为Department类型)按照'id','name'两个属性排序(优先级依次减弱)

import operator  # 导入operator 包,pip install operator

Departs = []  # 待排序列表


class Department:  # 自定义的元素
    def __init__(self, id, name, id2):
        self.id = id
        self.name = name
        self.id2 = id2


# 创建元素和加入列表
Departs.append(Department(1, 'cbc', '1'))
Departs.append(Department(2, 'acd', '4'))
Departs.append(Department(3, 'bcd', '1'))
Departs.append(Department(1, 'bcd', '1'))
Departs.append(Department(2, 'acd', '3'))

# 划重点#划重点#划重点----排序操作
cmpfun = operator.attrgetter('id2')  # 参数为排序依据的属性,可以有多个,这里优先id,使用时按需求改换参数即可
Departs.sort(key=cmpfun)  # 使用时改变列表名即可
# 划重点#划重点#划重点----排序操作

# 此时Departs已经变成排好序的状态了,排序按照id优先,其次是name,遍历输出查看结果
for depart in Departs:
    print(str(depart.id) + depart.name + depart.id2)

解释都在注释中,输出结果为:

 

 

可以看到

1.列表已经改变;

2.id和name都相同的两个部门的先后顺序未改变,说明该排序算法具有稳定性

一般性方法:

1. 首先导入operator包,pip install operator

2.关键操作为定义cmpfun 和使用.sort,即:

cmpfun = operator.attrgetter('a','b') #这里是希望按照a,b两个属性排序,且优先按照a属性,使用时更换参数即可。

list.sort(key=cmpfun)#list为待排序列表

推荐阅读