在 Python 中使用列表理解生成一个列表，它给出数字的计数并将数字放在特定的索引处

Question

假设我有一个列表

input = [4, 4, 2, 2, 2, 2, 3, 3, 3, 3, 1, 1, 2, 2, 2]

这个列表可以包含从 0 到 n 的（重复）数字（这里 n=4）。我可以使用列表推导生成列表吗，其中索引处的值等于列表 a 中该数字的计数例如，上述场景中的输出列表将是

output = [0, 2, 7, 4, 2]

说明

here
output[0] = 0 #since we don't have any 0 in input list
output[1] = 2 #since we have two 1's in input list
output[2] = 7 #since we have seven 2's in input list
output[3] = 4 #since we have four 3's in input list
output[4] = 2 #since we have two 4's in input list

Answer 1

您可以使用 collections.Counter:

from collections import Counter

inp = [4, 4, 2, 2, 2, 2, 3, 3, 3, 3, 1, 1, 2, 2, 2]

c = Counter(inp)
output = [c[k] for k in range(max(c)+1)]

注意。不要使用 input 作为变量名，这是一个 python 内置

输出：

[0, 2, 7, 4, 2]

Answer 2

d = {}
[d.setdefault(el, []).append(1) for el in input]
[len(d.get(k, [])) for k in range(max(d.keys()) + 1)]
# [0, 2, 7, 4, 2])

在 Python 中使用列表理解生成一个列表，它给出数字的计数并将数字放在特定的索引处

Generate a list using list comprehension in Python which gives the count of numbers and put the number at the specific index

python

list-comprehension