Question

我有一个列表列表，我想根据列表的第一个元素按升序排序。如果列表的第一个元素相同，则应根据第二个元素对它们进行排序。

到目前为止，我只能根据列表的第一个元素进行排序。我已经使用插入排序来对它们进行排序。如果第一个元素相同，我如何根据第二个元素对列表进行排序？

def sort_list ():
    # An example of the list to be sorted
    original_list = [['Glenn', 'Stevens'],
                    ['Phil', 'Wayne'],
                    ['Peter', 'Martin'],
                    ['Phil', 'Turville'],
                    ['Chris', 'Turville']]

    sorted_list = list(original_list)

    for index in range(1, len(sorted_list)):           
        pos = index                                 
        while pos > 0 and sorted_list[pos - 1][0] > sorted_list[pos][0]:    
            sorted_list[pos-1], sorted_list[pos] = sorted_list[pos], sorted_list[pos-1]
            pos -= 1                            

    return sorted_list

Answer 1

如果您想使用自己的功能进行排序，可以这样做。

要检查第二个元素，如果第一个元素相等，则只需写入

   (sorted_list[pos - 1][0] > sorted_list[pos][0] 
or (sorted_list[pos - 1][0] == sorted_list[pos][0] 
    and sorted_list[pos - 1][1] > sorted_list[pos][1]))

而不是

sorted_list[pos - 1][0] > sorted_list[pos][0]

实际上你可以写得更短：

sorted_list[pos - 1] > sorted_list[pos]

这正是你所需要的。

当python比较列表时，它会比较从第一个[0]开始的元素：

>>> a=[1,2]
>>> b=[1,1]
>>> a<b
False
>>> a=[1,2]
>>> b=[1,3]
>>> a<b
True
>>> a=[1,2]
>>> b=[2,1]
>>> a<b
True

Answer 2

列表比较已经按照你想要的方式工作（它被称为词汇顺序）：比较第一个项目，如果它们相等，则比较第二个和后续项目。

这意味着您可以使用一行对列表进行排序：

original_list.sort()

如果你必须实现自己的排序，你应该以一般方式实现它，传入一个关键函数（比如内置的排序函数）。

def insertion_sort(xs, key=(lambda x: x)):
    result = list(xs)
    for i in xrange(len(result)):
        for pos in xrange(i, 0, -1):
            if key(result[pos-1]) <= key(result[pos]):
                break
            result[pos-1], result[pos] = result[pos], result[pos-1]
    return result

现在您可以按每个子列表的第一个元素进行排序：

print insertion_sort(xs, key=(lambda x: x[0]))

或通过词汇顺序：

print insertion_sort(xs)

使用选择排序对列表进行排序

2 个答案: