如何根据列表中的键对元组列表进行排序？

Question

我想基于许多键对元组列表进行排序。 例如，我有这个元组列表：

list_t = [(1, 3, 5, 6, 9, 10), (1, 2, 3, 4, 5, 61), (1, 2, 3, 0, 9, 81), (1, 2, 6, 7, 9, 54), (1, 3, 5, 6, 12, 43)]

有一次我想通过元组的第一，第二，第三和第五元素对它进行排序：

keys = [0, 1, 2, 4]
list_t_sorted = [(1, 2, 3, 4, 5, 61), (1, 2, 3, 0, 9, 81), (1, 2, 6, 7, 9, 54), (1, 3, 5, 6, 9, 10), (1, 3, 5, 6, 12, 43)]

另一次我需要通过元组的第三个元素对它进行排序：

keys = [2]
list_t_sorted = [(1, 2, 3, 4, 5, 61), (1, 2, 3, 0, 9, 81), (1, 3, 5, 6, 9, 10), (1, 3, 5, 6, 12, 43), (1, 2, 6, 7, 9, 54)]

到目前为止，我尝试了这段代码，但它没有返回预期的结果：

def sort_list(keys, list_t):
    return sorted(list_t, key=lambda l: (l[x] for x in keys))

例如，对于keys = [0,1,2,4]，它返回[（1,3,5,6,9,10），（1,2,3,4,5,61），（1,3） ，5,6,12,43），（1,2,6,7,9,54），（1,2,3,0,9,81）]，它们不是基于键排序的。

有谁能够帮助我？ 谢谢！

Answer 1

干得好。 您可以在文档中阅读有关operator.itemgetter更多信息。

import operator
list_t = [(1, 3, 5, 6, 9, 10), (1, 2, 3, 4, 5, 61), (1, 2, 3, 0, 9, 81), (1, 2, 6, 7, 9, 54), (1, 3, 5, 6, 12, 43)]
keys = [0, 1, 2, 4]
sorted(list_t, key=operator.itemgetter(*keys))

输出：

[(1, 2, 3, 4, 5, 61),
 (1, 2, 3, 0, 9, 81),
 (1, 2, 6, 7, 9, 54),
 (1, 3, 5, 6, 9, 10),
 (1, 3, 5, 6, 12, 43)]

Answer 2

michael_j_ward的答案对你的问题来说是一个更清晰的解决方案，但如果你想知道你最初的想法出了什么问题：

您正在尝试对生成器对象进行排序，我猜测它具有不同的等同方式。 显示这一点的方法是通过显示您实际尝试使用for循环排序的内容：

>>> def sort_list(keys, list_t):
...     for l in list_t:
...             print (l[x] for x in keys)
...     return sorted(list_t, key = lambda l: (l[x] for x in keys))
...
>>> sort_list(keys, list_t)
<generator object <genexpr> at 0x105bc1af0>
<generator object <genexpr> at 0x105bc1af0>
<generator object <genexpr> at 0x105bc1af0>
<generator object <genexpr> at 0x105bc1af0>
<generator object <genexpr> at 0x105bc1af0>
[(1, 3, 5, 6, 12, 43), (1, 2, 6, 7, 9, 54), (1, 2, 3, 0, 9, 81), (1, 2, 3, 4, 5, 61), (1, 3, 5, 6, 9, 10)]

这显然不是您的预期输出，而是与这些生成器的比较方式有关。 如果你想用你最初的想法（没有itemgetter ）来做，那么显式创建tuples而不是生成器：

>>> def sort_list(keys, list_t):
...     return sorted(list_t, key = lambda l: tuple(l[x] for x in keys))
...
>>> sort_list(keys, list_t)
[(1, 2, 3, 4, 5, 61), (1, 2, 3, 0, 9, 81), (1, 2, 6, 7, 9, 54), (1, 3, 5, 6, 9, 10), (1, 3, 5, 6, 12, 43)]