如何在Python中为列表中的重复项添加计数?
这里有一个包含重复元素的列表:
l1 = ['a', 'b', 'c', 'a', 'a', 'b']
我们想要的结果是:
l1 = ['a', 'b', 'c', 'a_1', 'a_2', 'b_1']
我们该如何给重复的元素加上计数编号呢?
这是一个尝试实现这个目标的方法;不过,有没有更符合Python风格的做法呢?
for index in range(len(l1)):
counter = 1
list_of_duplicates_for_item = [dup_index for dup_index, item in enumerate(l1) if item == l1[index] and l1.count(l1[index]) > 1]
for dup_index in list_of_duplicates_for_item[1:]:
l1[dup_index] = l1[dup_index] + '_' + str(counter)
counter = counter + 1
5 个回答
1
根据你对@mathmike的评论,如果你的最终目标是从一个有重复键的列表中创建一个字典,我建议你使用`collections`库中的defaultdict
。
>>> from collections import defaultdict
>>> multidict = defaultdict(list)
>>> multidict['a'].append(1)
>>> multidict['b'].append(2)
>>> multidict['a'].append(11)
>>> multidict
defaultdict(<type 'list'>, {'a': [1, 11], 'b': [2]})
5
我会这样做:
a1 = ['a', 'b', 'c', 'a', 'a', 'b']
a2 = []
d = {}
for i in a1:
d.setdefault(i, -1)
d[i] += 1
if d[i] >= 1:
a2.append('%s_%d' % (i, d[i]))
else:
a2.append(i)
print a2
20
在Python中,创建一个新的列表通常比修改一个已有的列表要简单得多。我们可以使用生成器来高效地完成这个任务。同时,字典可以用来记录某些东西出现的次数。
l = ['a', 'b', 'c', 'a', 'a', 'b']
def rename_duplicates( old ):
seen = {}
for x in old:
if x in seen:
seen[x] += 1
yield "%s_%d" % (x, seen[x])
else:
seen[x] = 0
yield x
print list(rename_duplicates(l))