Question

说我有一个列表，其中包含一些重复的数字。

list = [1,1,1,1,2,3,4,4,1,2,5,6]

我想识别列表中重复和连续的所有元素，包括第一个元素，即将列表中的所有元素替换为字典中的值：

mydict = {1: 'a', 4: 'd'}    
list = ['a','a','a','a',2,3,'d','d',1,2,5,6]

因为我也想替换重复的第一个实例，所以我很困惑如何继续！

Answer 1

itertools.groupby是你的朋友：

from itertools import groupby

mydict = {1: 'a', 4: 'd'}
A = [1,1,1,1,2,3,4,4,1,2,5,6]

res = []
for k, g in groupby(A):
  size = len(list(g))
  if size > 1:
    res.extend([mydict[k]] * size)  # see note 1
  else:
    res.append(k)
print(res)  # -> ['a', 'a', 'a', 'a', 2, 3, 'd', 'd', 1, 2, 5, 6]

注意：

如果您希望捕获可能的KeyError并且您希望重新使用的默认值，请使用mydict.get(k, <default>)代替mydict[k]

替换列表中连续且在Python中重复的所有元素

1 个答案: