如何为列表中的每个唯一val添加计数

Question

假设我有一个清单.

temp = ['A', 'B', 'A', 'B', 'A', 'B']

我正在寻找一种方法来加入里面的字符串计数.

预期产出:

['A_1', 'B_1', 'A_2', 'B_2', 'A_3', 'B_3']

我能够通过使用列表解析来解决它,但我正在寻找一种方法,我不必指定列表[1, 1, 2, 2, 3, 3].可能吗？

[j + "_" + str(i) for i, j in zip([1, 1, 2, 2, 3, 3], temp)]

Answer 1

您可以使用collections.defaultdict一个for循环:

from collections import defaultdict

L = ['A', 'B', 'A', 'B', 'A', 'B']

dd = defaultdict(int)

res = []
for item in L:
    dd[item] += 1
    res.append(f'{item}_{dd[item]}')

print(res)

['A_1', 'B_1', 'A_2', 'B_2', 'A_3', 'B_3']

Answer 2

您可以使用a Counter或a defaultdict(int)来跟踪角色在遇到角色时被看到的次数.

>>> from collections import Counter
>>> 
>>> temp = ['A', 'B', 'A', 'B', 'A', 'B']
>>> seen = Counter()
>>> 
>>> result = []
>>> for c in temp:
...:    seen.update(c)
...:    result.append('{}_{}'.format(c, seen[c]))
...:    
>>> result
>>> ['A_1', 'B_1', 'A_2', 'B_2', 'A_3', 'B_3']

请注意,seen.update(c)如果您希望字符串中包含多个字符,则可能会产生意外结果temp.演示:

>>> seen = Counter()
>>> seen.update('ABC')
>>> seen
>>> Counter({'A': 1, 'B': 1, 'C': 1})

根据您的计算方式以及您期望的数据类型,您可能希望使用该行

seen[c] += 1

代替

seen.update(c)

或者,没有任何进口:

>>> seen = {}
>>> result = []
>>> 
>>> for c in temp:
...:    seen[c] = seen.get(c, 0) + 1
...:    result.append('{}_{}'.format(c, seen[c]))
...:    
>>> result
>>> ['A_1', 'B_1', 'A_2', 'B_2', 'A_3', 'B_3']