如何根据另一个列表对列表进行排序？

Question

有一个清单：a = [("ax", 1), ("ec", 3), ("bk", 5)]另一个清单：b = ["ec", "ax", "bk"]我要排序a根据b：sort_it(a, b)a = [("ec", 3), ("ax", 1), ("bk", 5)]这个怎么做？

Answer 1

a.sort(key=lambda x: b.index(x[0]))

这a使用b每个元组的第一个元素的index in a作为对其进行排序的值就地进行排序。

另一种可能更简洁的书写方式是：

a.sort(key=lambda (x,y): b.index(x))

如果您有大量项目，则做一些不同的事情可能会更有效率，因为.index()在长列表上可能是一项昂贵的操作，并且由于您已经知道顺序，因此实际上不需要进行完全排序：

mapping = dict(a)

a[:] = [(x,mapping[x]) for x in b]

请注意，这仅适用于2元组的列表。如果您希望它适用于任意长度的元组，则需要对其进行一些修改：

mapping = dict((x[0], x[1:]) for x in a)

a[:] = [(x,) + mapping[x] for x in b]

Answer 2

实际上，有一种方法可以在线性O（n）时间中执行此操作，因为这实际上不是排序操作。列表的存在b意味着排序已经完成；我们真正需要做的就是重新排列元素的a顺序。多亏了词典，这可以有效地完成。

from collections import defaultdict

def sorted_by(seq_to_sort, desired_order, key=None):

    if key is None:

        key = lambda x: x

    # group the elements by their key

    grouped_items = defaultdict(list)

    for item in seq_to_sort:

        k = key(item)

        grouped_items[k].append(item)

    # flatten the dict of groups to a list

    return [item for key in desired_order for item in grouped_items[key]]

用法：

a = [("ax", 1), ("ec", 3), ("bk", 5)]

b = ["ec", "ax", "bk"]

result = sorted_by(a, b, lambda tup: tup[0])

print(result)  # output: [("ec", 3), ("ax", 1), ("bk", 5)]

笔记：

这是一个稳定的排序；如果两个列表项具有相同的键，则将保留其顺序。例：

>>> sorted_by([1, 2, 3], [5], key=lambda x: 5)

[1, 2, 3]

如果有任何列表元素映射到中不存在的键，则desired_order这些元素将被静默丢弃。例如：

>>> sorted_by([1, 2, 3], [1, 2, 3], key=lambda x: 5)

[]

也可以看看：

defaultdict

Answer 3

另一个posibility是进行排序a，对排序索引b根据b与比所述排序a根据索引

a.sort(key=lambda x: x[0])

ind = [i[0] for i in sorted(enumerate(b),key=lambda x: x[1])]

a = [i[0] for i in sorted(zip(a,ind),key=lambda x: x[1])]

由于每次排序都需要n * log（n），因此对于较大的列表仍然可以扩展