python - 按频率和值对列表进行排序

Question

我正在尝试解决以下问题:函数接受列表 A。结果必须是列表的有序列表。每个列表包含与原始列表A中频率相同的元素。

例子:

Input: [3, 1, 2, 2, 4]
Output: [[1, 3, 4], [2, 2]]

我设法对初始列表 A 进行排序并确定元素出现的频率。但是，我不知道如何根据频率拆分原始列表 A。

我的代码:

def customSort(arr):
    counter = Counter(arr)
    y = sorted(arr, key=lambda x: (counter[x], x))
    print(y)
    x = Counter(arr)
    a = sorted(x.values())
    print()

customSort([3,1,2,2,4])

我当前的输出:

[1, 3, 4, 2, 2]
[1, 1, 1, 2]

Answer 1

您可以使用列表的 defaultdict 并迭代您的 Counter:

from collections import defaultdict, Counter

def customSort(arr):
    counter = Counter(arr)
    dd = defaultdict(list)
    for value, count in counter.items():
        dd[count].extend([value]*count)
    return dd

res = customSort([3,1,2,2,4])
# defaultdict(list, {1: [3, 1, 4], 2: [2, 2]})

这提供了额外的信息，即键表示列表中的值被看到的次数。如果您需要列表的列表，您可以简单地访问值:

res = list(res.values())
# [[3, 1, 4], [2, 2]]