快捷導(dǎo)航

Python sort 自定義函數(shù)排序問(wèn)題

更新時(shí)間：2023年09月12日 09:17:39 作者：世界盡頭與你

這篇文章主要介紹了Python sort 自定義函數(shù)排序問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

Python sort 自定義函數(shù)排序

sort() 函數(shù)用于對(duì)原列表進(jìn)行排序，如果指定參數(shù)，則使用比較函數(shù)指定的比較函數(shù)。

sort() 方法語(yǔ)法：

list.sort( key=None, reverse=False)

參數(shù)

key – 主要是用來(lái)進(jìn)行比較的元素，只有一個(gè)參數(shù)，具體的函數(shù)的參數(shù)就是取自于可迭代對(duì)象中，指定可迭代對(duì)象中的一個(gè)元素來(lái)進(jìn)行排序。
reverse – 排序規(guī)則， reverse = True 降序， reverse = False 升序（默認(rèn)）。

以下實(shí)例演示了通過(guò)指定列表中的元素排序來(lái)輸出列表：指定第二個(gè)元進(jìn)行素排序

def take_second(elem):
    return elem[1]
random = [(2, 2), (3, 4), (5, 6), (1, 5), (6, 3)]
random.sort(key=take_second)
print(random)  # [(2, 2), (6, 3), (3, 4), (1, 5), (5, 6)]

自定義Python排序函數(shù)比較方式

當(dāng)你想按自己的方式對(duì)數(shù)組元素進(jìn)行排序時(shí)，我們需要自定義比較函數(shù)實(shí)現(xiàn)我們想實(shí)現(xiàn)的排序方式。

例1

以降序?qū)?shù)組進(jìn)行排序

>>> def comp(x,y):
...     return y-x
... 
>>> a = [1,8,4,5,2,7]
>>> a.sort(comp)
>>> a
[8, 7, 5, 4, 2, 1]

例2

class Interval(object):
	def __init__(self, s=0, e=0):
		self.start = s
		self.end = e
intervals = []
tmp = [[1,3],[15,18],[8,10],[2,6]]
for l in tmp:
	t = Interval(l[0],l[1])
	intervals.append(t)
def printInterval(inter):
	print("[{}, {}]".format(inter.start,inter.end))
for l in intervals:
	printInterval(l)
# [[1,3],[15,18],[8,10],[2,6]]
def comp(it1, it2):
	return it1.start-it2.start
intervals.sort(comp)
for l in intervals:
	printInterval(l)
# [[1,3],[2,6],[8,10],[15,18]]

以上方法皆是python2的實(shí)現(xiàn)。python3下實(shí)現(xiàn)有所不同（以上方法python3下跑不通），因?yàn)閜ython3 sorted取消了對(duì)cmp的支持，即python3沒(méi)有cmp函數(shù)。

例2的python3實(shí)現(xiàn)

class Interval(object):
	def __init__(self, s=0, e=0):
		self.start = s
		self.end = e
intervals = []
tmp = [[1,3],[15,18],[8,10],[2,6]]
for l in tmp:
	t = Interval(l[0],l[1])
	intervals.append(t)
def printInterval(inter):
	print("[{}, {}]".format(inter.start,inter.end))
for l in intervals:
	printInterval(l)
def comp(it1):
	return it1.start
intervals.sort(key=comp)
# sorted(intervals,key=comp)
for l in intervals:
	printInterval(l)

在python3中，待比較元素x通過(guò)custom_key函數(shù)轉(zhuǎn)化為Python能比較的值custom_key(x)，進(jìn)而再基于返回值進(jìn)行排序。

python3 比較第二種方式

如果想和python2中一樣定義比較函數(shù)，可以通過(guò)functools庫(kù)中的cmp_to_key()函數(shù)將比較函數(shù)(comparison function)轉(zhuǎn)化為關(guān)鍵字函數(shù)（key function)。

functools.cmp_to_key(func)函數(shù)將比較函數(shù)轉(zhuǎn)化為關(guān)鍵字函數(shù)，與接受key function 的工具一同使用（如 sorted(), min(), max())。

from functools import cmp_to_key as ctk
class Interval(object):
	def __init__(self, s=0, e=0):
		self.start = s
		self.end = e
intervals = []
tmp = [[1,3],[15,18],[8,10],[2,6]]
for l in tmp:
	t = Interval(l[0],l[1])
	intervals.append(t)
def printInterval(inter):
	print("[{}, {}]".format(inter.start,inter.end))
for l in intervals:
	printInterval(l)
"""
Please note here.
"""
def comp(it1,it2):
	return it1.start-it2.start
intervals.sort(key=ctk(comp))
# sorted(intervals,key=comp)
for l in intervals:
	printInterval(l)

最后得到的結(jié)果和上面的代碼是一樣的。