如何按子列表的第一项然后按第三项

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],
...
]

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的建议。