1. 程式人生 > 程式設計 >Python 使用多屬性來進行排序

Python 使用多屬性來進行排序

Python 中 list.sort() 是列表中非常常用的排序函式, key 引數可以對單個屬性進行排序。

但是想要實現類似 sql 中 order by id,age 一樣,對多個欄位進行排序就不支援了。

py2 中 sort() 函式還有個 cmp 引數可以傳入一個方法,可以自定義對多個屬性進行排序,py3 中移除了這個欄位。

py3 想要實現這個功能,需要使用 functools 模組中的方法,例項如下

#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author: wxnacy([email protected])

from functools import cmp_to_key

def cmp_func(a,b):

  if a['id'] < b['id']:
    return -1
  elif a['id'] > b['id']:
    return 1

  if a['name'] < b['name']:
    return -1
  elif a['name'] > b['name']:
    return 1

  return 0

arr.sort(key=cmp_to_key(cmp_func))

上面的例子我們實現了 id 正序排序,如果 id 相同,則按照 age 正序排序

為了方便可以封裝成一個方法,以供列表呼叫,封裝的方法 demo 可以參見地址

https://github.com/wxnacy/study/blob/master/python/office_module/list_demo/sort_by_multi_fields.py

總結

以上所述是小編給大家介紹的Python 使用多屬性來進行排序,希望對大家有所幫助,如果大家有任何疑問歡迎給大家留言,小編會及時回覆大家的!