python - 如何使用快速排序在列表列表中按索引排序

Question

我正在尝试使用快速排序对索引 [1] 处的列表列表进行排序。例如:

list = [[2, 5, 3],
        [2, 4, 9],
        [0, 9, 1],
        [1, 1, 1],
        [4, 7, 5]]

我希望能够做到这一点，按索引[1]排序:

quickSort(list)

output:

list = [[1, 1, 1],
        [2, 4, 9],
        [2, 5, 3],
        [4, 7, 5],
        [0, 9, 1]]

有什么想法吗？

Answer 1

您想要在 list.sort 中使用 key 参数:

import operator
mylist = [[2, 5, 3],
          [2, 4, 9],
          [0, 9, 1],
          [1, 1, 1],
          [4, 7, 5]]

mylist.sort(key=operator.itemgetter(1))

输出:

>>> mylist = [[2, 5, 3],
...           [2, 4, 9],
...           [0, 9, 1],
...           [1, 1, 1],
...           [4, 7, 5]]
>>> 
>>> mylist.sort(key=operator.itemgetter(1))
>>> mylist
[[1, 1, 1], [2, 4, 9], [2, 5, 3], [4, 7, 5], [0, 9, 1]]