0

我在一本书中看到如何将特定的排序函数传递给 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函数添加第三个参数以用于不同的排序类型？

谢谢！

4

2 回答 2

5

您真的想改用该key参数；在第 4 列排序operator.itemgetter()：

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 个选项都创建了一个新的可调用对象，该可调用对象在data.

cmp参数 tosorted()已在 Python 3 中删除。

于 2013-04-18T22:04:19.353 回答
3

您通常不使用cmp（第二个参数）进行排序。该key论点在 99% 的情况下都是最佳选择：

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))
于 2013-04-18T22:05:35.623 回答