python - 如何按子列表的第一项然后按第三项对列表列表进行排序

Question

我有一个列表列表。例如，

[ 
[1, 2, 0.10], 
[1, 3, 0.22],
[1, 4, 0.20],
[1, 5, 0.37],
[1, 8, 0.60],
[1, 10, 0.01],
[1, 15, 0.26],
[2, 3, 0.28],
[2, 6, 0.12],
[2, 7, 0.72],
[2, 10, 0.77],
[2, 11, 0.36],
...
]

我想按子列表的第一项然后按第三项对列表列表进行排序，并获得每个第一项的前三个结果，该怎么做？谢谢！

结果示例，

[ 
[1, 15, 0.26],
[1, 5, 0.37],
[1, 8, 0.60],
[2, 11, 0.36],
[2, 7, 0.72],
[2, 10, 0.77],
...
]

标签： pythonlistsorting

Answer 1

这是使用itertools.groupby. 我们使用一个tuple键按第一项排序，然后反向排序最后一项。

from itertools import groupby

sorter = sorted(A, key=lambda x: (x[0], -x[2]))
grouper = groupby(sorter, key=lambda x: x[0])

res = [list(j)[:3] for _, j in grouper]

print(res)

[[[1, 8, 0.6], [1, 5, 0.37], [1, 15, 0.26]],
 [[2, 10, 0.77], [2, 7, 0.72], [2, 11, 0.36]]]

看来您想先降序排序，然后再升序排序。如果这是一个要求，您可以反转结果，即

res = [list(j)[:3][::-1] for _, j in grouper]

感谢@Ev.Kounis 的建议。