10

背景：

我想知道如何实现高级排序函数，这些函数可以作为元组元素传递给 python 'sorted' 函数的 key 参数。

这是一个描述我想做的例子：

class Book:

      def __init__(self, name, author, language, cost):
          self.name = name
          self.author = author
          self.language=language
          self.cost = cost


bookList = [list of books]

firstLanguage = "Armenian"
possibleLanguages = ["English", "Spanish", "Armenian", "French", "Chinese", "Swahili"]
possibleLanguages.remove("Armenian")

sortedBookList = sorted(bookList, key=(sortByName,
    sortByFirstLanguage(firstLanguage), sortByLanguages(possibleLanguages) ))

基本上我想实现上面描述的'sortByFirstLanguage'和'sortByLanguages'函数，以便我可以将它们作为'key'参数的元组项传递给python'sorted'函数。下面是一些关于自定义排序函数应该是什么样子的示例代码：

def sortByName(elem):
    return elem.name

def sortByFirstLanguage(elem, firstLanguage):
    if elem.language == firstLanguage:
       return 1
    else:
       return -1


def sortByLanguages(elem, possibleLanguages):
    if elem.language in possibleLanguages:
       return possibleLanguages.index(elem.language)

添加。细节：

  1. 我正在使用 python 2.7
  2. 这个问题实际上是使用 Django 查询集而不是对象列表，但为了演示目的，我认为对象列表具有相同的目的。
  3. 这种排序的目标是首先按指定的语言排序，然后返回 && 按照默认排序（在本例中为列表排序）对剩余项目进行排序。

问题：

如上所示，我如何准确地告诉“关键”参数将额外的参数“firstLanguage”&&“possibleLanguages”传递给自定义排序函数？

4

1 回答 1

7

正如 Ashish 在评论中指出的那样，我们首先需要组合这些函数，因为key只接受一个函数。如果我们返回函数结果的序列 (list, tuple)，Python 会做正确的事情，如果前面的元素相等 ( source )，则只比较后面的（更右边的）元素。

我知道有几种方法可以做到这一点。

使用 lambda：

sortedBookList = sorted(
    bookList, 
    key=lambda elem: (sortByName(elem), 
                      sortByFirstLanguage(elem, firstLanguage), 
                      sortByLanguages(elem, possibleLanguages)))

使用高阶函数：

def key_combiner(*keyfuncs):
  def helper(elem):
    return [keyfunc(elem) for keyfunc in keyfuncs]
  return helper

def sortByFirstLanguage(firstLanguage):
  def helper(elem):
    return elem.language == firstLanguage  # True > False
  return helper

def sortByLanguages(possibleLanguages):
  def helper(elem):
    if elem.language in possibleLanguages:
       return possibleLanguages.index(elem.language)
  return helper

sortedBookList = sorted(bookList,
                        key=key_combiner(sortByName, 
                                         sortByFirstLanguage(firstLanguage), 
                                         sortByLanguages(possibleLanguages))

Lambdas 对我来说似乎是最干净的，所以这可能是我会使用的。

于 2013-10-27T20:21:35.887 回答