2

您将如何lista根据以下项目的顺序进行排序sorter_list：

lista = [["John", "B3"],["Robert", "P3"], ["Thomas", "S2"]]

sorter_list = ["P3", "S2", "B3"]

结果将是：

sorted_lista = [ ["Robert", "P3"], ["Thomas", "S2"], ["John", "B3"]]

问候

4

5 回答 5

7

假设总是有一个条目sorter_list匹配 中每个列表的第二个元素lista：

sorted_lista = sorted(lista, key=lambda lst: sorter_list.index(lst[1]))
于 2013-09-12T17:00:08.103 回答
3

虽然@FJ 有一个完美的解决方案，但我的问题是，你为什么不首先使用字典来存储这种数据？

带字典：

d = {'B3': 'John', 'P3': 'Robert', 'S2': 'Thomas'}
sorter = ["P3", "S2", "B3"]
print([(d[key], key) for key in sorter])

输出：

[('Robert', 'P3'), ('Thomas', 'S2'), ('John', 'B3')]

另外：您还应该检查collections模块的OrderedDict.

更新：

当然，您可以将值存储为列表，因此可以保存多个值：

带字典：

d = {'B3': [('John', 123)], 'P3': [('Robert', 465), ('Andres', 468)], 'S2': [('Thomas', 19)]}
sorter = ('P3', 'B3', 'S2')
print([(d[key], key) for key in sorter])

输出：

[([('Robert', 465), ('Andres', 468)], 'P3'), ([('John', 123)], 'B3'), ([('Thomas', 19)], 'S2')]

在这种情况下，您还可以在字典中使用字典：

d = {'B3': {'John': 123}, 'P3': {'Robert': 465, 'Andres': 468}, 'S2': {'Thomas': 19}}

以后查找会容易得多。

于 2013-09-12T17:05:58.787 回答
1

我是python的新手，所以它可能不是最优化的解决方案

sorted_lista=[]

for i in sorter_list:
    for j in lista:
        if i==j[1]:
            sorted_lista.append([j[0],j[1]])

print sorted_lista

输出 ：

[['Robert', 'P3'], ['Thomas', 'S2'], ['John', 'B3']]
于 2013-09-12T17:08:53.317 回答
1

您可以通过构建字典在 O(N) 中执行此操作，其中您的键是B3等S2。

lookup_dict = dict( (item[1],item) for item in lista)
sorted_lista = [ lookup_dict[key] for key in sorter_list ]

这利用了您sorter_list已经排序的事实。

于 2013-09-12T17:04:50.190 回答
1

为了高效排序，我认为最好从sorter_list

sorter_dict = {x:i for i, x in enumerate(sorter_list)}
sorted_lista = sorted(lista, key=lambda lst: sorter_dict[lst[1]])
于 2013-09-12T17:06:12.753 回答