python - 返回每组具有最高值的列表

Question

我目前有一个位置列表，我想整理一下。

该列表如下所示:

list = [['Location 1', 5],['Location 2', 5],['Location 3', 5],['Location 1', 4],['Location 2', 6],['Location 3', 5],['Location 1', 5],['Location 2', 5]]

目标是为每个位置选择索引 1 中每个列表的最高值。最终结果应如下所示:

correctList = [['Location 1', 5],['Location 2', 6],['Location 3', 5]]

具有相同整数值的位置没有优先权。

我现在拥有的解决方案是根据名称将每个位置附加到自己的列表中。然后从每个列表中对每个位置列表使用 max() 操作。

Answer 1

您可以使用collections.defaultdict对于 O(n) 解决方案:

from collections import defaultdict

L = [['Location 1', 5],['Location 2', 5],['Location 3', 5],['Location 1', 4],
     ['Location 2', 6],['Location 3', 5],['Location 1', 5],['Location 2', 5]]

dd = defaultdict(int)

for location, value in L:
    dd[location] = max(dd[location], value)

print(dd)
# defaultdict(int, {'Location 1': 5, 'Location 2': 6, 'Location 3': 5})

这给出了字典映射。如果您热衷于列表列表:

res = list(map(list, dd.items()))

print(res)
# [['Location 1', 5], ['Location 2', 6], ['Location 3', 5]]