一个例子:
names = ["George Washington", "John Adams", "Thomas Jefferson", "James Madison"]
sorted(names, key=lambda name: name.split()[-1].lower())
我知道key
用于比较不同的名称,但它可以有两种不同的实现:
第一种方法的问题是它必须定义另一个数据结构来绑定密钥和数据。第二种方法的问题是密钥可能被多次计算,即name.split()[-1].lower()
将被执行多次,这非常耗时。
我只是想知道Python实现sorted()
的方式。
答案 0 :(得分:5)
关键功能仅按执行一次每个值,以产生(keyvalue, value)
对;然后将其用于排序,稍后将按排序顺序返回值。这有时称为Schwartzian transform。
你可以自己测试一下;您可以计算调用函数的频率,例如:
>>> def keyfunc(value):
... keyfunc.count += 1
... return value
...
>>> keyfunc.count = 0
>>> sorted([0, 8, 1, 6, 4, 5, 3, 7, 9, 2], key=keyfunc)
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> keyfunc.count
10
或者您可以收集传递的所有值;您会看到他们遵循原始输入顺序:
>>> def keyfunc(value):
... keyfunc.arguments.append(value)
... return value
...
>>> keyfunc.arguments = []
>>> sorted([0, 8, 1, 6, 4, 5, 3, 7, 9, 2], key=keyfunc)
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>> keyfunc.arguments
[0, 8, 1, 6, 4, 5, 3, 7, 9, 2]
如果要读取CPython源代码,则相关函数称为listsort()
,keyfunc
用于以下循环(saved_ob_item
是输入数组),在排序发生之前执行:
for (i = 0; i < saved_ob_size ; i++) {
keys[i] = PyObject_CallFunctionObjArgs(keyfunc, saved_ob_item[i],
NULL);
if (keys[i] == NULL) {
for (i=i-1 ; i>=0 ; i--)
Py_DECREF(keys[i]);
if (saved_ob_size >= MERGESTATE_TEMP_SIZE/2)
PyMem_FREE(keys);
goto keyfunc_fail;
}
}
lo.keys = keys;
lo.values = saved_ob_item;
所以最后,你有两个数组,一个有keys
,另一个有原始值。所有排序操作并行处理两个数组,对lo.keys
中的值进行排序并同时移动lo.values
中的元素。