python - 获取列表中所有元素的增量计数

Question

我有一个包含 2400 万个元素的列表，我想迭代地增加每个元素的计数，并以更快的方式将计数存储在另一个列表中。例如，我的列表是:

a=['自行车','自行车','吉普车','马','马','马','飞行','飞行','自行车']

我的预期输出是

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

我使用的代码是

z=[]
for i in a:
   z.append(a.count(i))

但是我的输出有点不同

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

我对这个新创建列表的顺序也很重要，应该基于我的列表(a)。非常感谢任何帮助。

Answer 1

根据您的预期输出，由于您需要直到您在该时间点迭代的列表索引为止的元素计数，因此以下代码应该有效:

from collections import defaultdict

a=['bike','bike','jeep','horse','horse','horse','flight','flight','cycle']

a_dict = defaultdict(int)
a_output = []

for x in a:
    a_dict[x] += 1
    a_output.append(a_dict[x])

print(a_output)

输出:

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