Question

我在书中看到如何将特定的排序函数传递给Python自己的内置sorted（）函数，如下所示：

def mysort(a, b):
    if a[3] < b[3]:
        return -1
    elif a[3] > b[3]:
        return 1
    else:
        return 0

data = [
('Alpha Centauri A', 4.3, 0.26, 1.56),
('Alpha Centauri B', 4.3, 0.077, 0.45),
('Alpha Centauri C', 4.2, 0.00001, 0.00006),
("Barnard's Star", 6.0, 0.00004, 0.0005),
('Wolf 359', 7.7, 0.000001, 0.00002),
('BD +36 degrees 2147', 8.2, 0.0003, 0.006),
('Luyten 726-8 A', 8.4, 0.000003, 0.00006),
('Luyten 726-8 B', 8.4, 0.000002, 0.00004),
('Sirius A', 8.6, 1.00, 23.6),
('Sirius B', 8.6, 0.001, 0.003),
('Ross 154', 9.4, 0.00002, 0.0005),
]

sorted_data = sorted(data, mysort)

上面的代码根据4元素元组的第4个元素对数据进行排序。在这里，我试图找出sorted（）函数如何将a和b参数提供给mysort函数。我的意图是将另一个参数传递给mysort函数，类似于：

def mysort(a, b, i):
    if a[i] < b[i]:
        return -1
    elif a[i] > b[i]:
        return 1
    else:
        return 0

它将告诉函数排序应该基于哪个元素。我很困惑，因为在行

sorted_data = sorted(data, mysort)

我们不会将任何参数传递给mysort函数。 sorted()函数似乎有自己的魔力，并为a函数提供b和mysort参数。总结一下，我想知道是否有办法为不同排序类型的mysort函数添加第三个参数？

谢谢！

Answer 1

你真的想要使用key参数;使用operator.itemgetter()在第4列进行排序：

from operator import itemgetter

sorted(data, key=itemgetter(3))

或者您可以使用lambda：

sorted(data, key=lambda elem: elem[3])

或者您可以使用functools.partial()：

from functools import partial

def mykeyfunc(column, item):
    return item[column]

sorted(data, key=partial(mykeyfunc, 3))

所有3个选项都会创建一个 new callable，并传递给data中的每个项目。

在Python 3中删除了cmp的{{1}}参数。

Answer 2

您通常不会使用cmp（第二个参数）进行排序。在{99}的情况下，key参数是最佳选择：

def mysort(item):
    return item[3]

sorted_data = sorted(data, key=mysort)

或者更简洁：

sorted_data = sorted(data, key=lambda item: item[3])

要使第二个功能起作用，您需要使用您的功能创建一个功能：

def mysort(i):
    def sort_func(a, b)
        if a[i] < b[i]:
            return -1
        elif a[i] > b[i]:
            return 1
        else:
            return 0

    return sort_func

并使用它：

sorted(data, mysort(3))

但更好的方法是使用内置的东西：

from operator imoprt itemgetter

sorted_data = sorted(data, key=itemgetter(3))

Python内置的sorted（）函数

2 个答案: