python - 如何使用相同的方法按不同的属性快速排序对象列表？

Question

我的类(class)如下:

class Member:
  def __init__(self, name, zip, hire_date, birth_date):
    self.id = id
    self.name = name
    self.zip = zip
    self.hire_date = hire_date
    self.birth_date = birth_date

  def get_name(self):
    return self.name

  def get_birth_date(self):
    return self.birth_date

  def get_hire_date(self):
    return self.hire_date

  def get_zip_code(self):
    return self.zip

我需要根据邮政编码、姓名、出生日期和雇用日期对此类对象的列表进行快速排序。我可以使用单独的排序方法来实现这些。例如，按邮政编码快速排序:

def partition(array, begin, end):
    pivot = begin
    for i in range(begin+1, end+1):
        if array[i].get_zip_code() <= array[begin].get_zip_code():
            pivot += 1
            array[i], array[pivot] = array[pivot], array[i]
    array[pivot], array[begin] = array[begin], array[pivot]
    return pivot



def quicksort(array, begin=0, end=None):
    if end is None:
        end = len(array) - 1
    def _quicksort(array, begin, end):
        if begin >= end:
            return
        pivot = partition(array, begin, end)
        _quicksort(array, begin, pivot-1)
        _quicksort(array, pivot+1, end)
    return _quicksort(array, begin, end)

这很好用。但是现在我如何使相同的方法适用于所有属性；理想情况下，通过将附加参数传递给 quicksort() 来定义要排序的属性。我也希望不直接调用对象属性，而是使用 getter 方法。

这是按 zip 排序后的输出

Name    Zip Hire Date   Birth Date
Cosmo Kramer 111-222 2019-06-01 1979-08-12
George Costanza 131-212 2016-06-01 1980-08-12
Jerry Seinfeld 223-212 2017-06-01 1981-06-15
John Wayne 323-212 2018-06-01 1962-06-17

Answer 1

添加另一个参数指示要排序的 key，或者只使用内置的 sort() 方法来做同样的事情:

class Member:
    def __init__(self, name, zip, hire_date, birth_date):
        self.id = id
        self.name = name
        self.zip = zip
        self.hire_date = hire_date
        self.birth_date = birth_date

    def get_name(self):
        return self.name

    def get_birth_date(self):
        return self.birth_date

    def get_hire_date(self):
        return self.hire_date

    def get_zip_code(self):
        return self.zip

    # Define how to print a Member
    def __str__(self):
        return f'{self.name:15} {self.zip} {self.hire_date} {self.birth_date}'

def partition(array, key, begin, end):
    pivot = begin
    for i in range(begin+1, end+1):
        if key(array[i]) <= key(array[begin]): # use key for comparison
            pivot += 1
            array[i], array[pivot] = array[pivot], array[i]
    array[pivot], array[begin] = array[begin], array[pivot]
    return pivot

def quicksort(array, key, begin=0, end=None):
    if end is None:
        end = len(array) - 1
    def _quicksort(array, key, begin, end):
        if begin >= end:
            return
        pivot = partition(array, key, begin, end)
        _quicksort(array, key, begin, pivot-1)
        _quicksort(array, key, pivot+1, end)
    return _quicksort(array, key, begin, end)

def display(L):
    print('Name            Zip     Hire Date  Birth Date')
    for item in L:
        print(item)
    print()

L =[Member('Cosmo Kramer','111-222','2019-06-01','1979-08-12'),
    Member('George Costanza','131-212','2016-06-01','1980-08-12'),
    Member('Jerry Seinfeld','223-212','2017-06-01','1981-06-15'),
    Member('John Wayne','323-212','2018-06-01','1962-06-17')]

quicksort(L,Member.get_hire_date)
display(L)
quicksort(L,Member.get_birth_date)
display(L)
# Using built-in sort()
L.sort(key=Member.get_name)
display(L)

输出:

Name            Zip     Hire Date  Birth Date
George Costanza 131-212 2016-06-01 1980-08-12
Jerry Seinfeld  223-212 2017-06-01 1981-06-15
John Wayne      323-212 2018-06-01 1962-06-17
Cosmo Kramer    111-222 2019-06-01 1979-08-12

Name            Zip     Hire Date  Birth Date
John Wayne      323-212 2018-06-01 1962-06-17
Cosmo Kramer    111-222 2019-06-01 1979-08-12
George Costanza 131-212 2016-06-01 1980-08-12
Jerry Seinfeld  223-212 2017-06-01 1981-06-15

Name            Zip     Hire Date  Birth Date
Cosmo Kramer    111-222 2019-06-01 1979-08-12
George Costanza 131-212 2016-06-01 1980-08-12
Jerry Seinfeld  223-212 2017-06-01 1981-06-15
John Wayne      323-212 2018-06-01 1962-06-17