python - 在 Python 中使用列表理解创建按预定义值排序的二维列表

Question

我有一个函数 number_sort()，它需要返回一个包含 4 个列表的二维列表。函数需要在 0-12、13-25、26-38 和 39-51 之间对给定的数字进行排序，如下所示:

给定: [2, 2, 3, 4, 1, 13, 15, 51, 10, 15, 28]

返回: [[1, 2, 2, 3, 4, 10], [13, 15, 15], [28], [51]]

我试着用列表推导来做。

def number_sort(cards: list):
    return [[item for item in cards if i * 13 <= item <= (i * 12 + 1)] for i in range(4)]

但是当我给出上面给出的确切示例列表时，它返回 [[1], [13], [], []]

谁能告诉我我做错了什么，我该如何解决？

Answer 1

IIUC，您想使用 13 作为楼层分隔符进行分组:

from itertools import groupby

[list(g) for _, g in groupby(sorted(l), key=lambda x: x//13)]

输出:

[[1, 2, 2, 3, 4, 10], [13, 15, 15], [28], [51]]