时间戳

首页 > 解决方案 > 在 Python 中替换列表中连续和重复的所有元素

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

问题描述

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

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]

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

标签： pythonpython-3.xlistduplicates

解决方案

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]

笔记：

  1. 如果你想捕获可能KeyError的 s 并有一个你想依赖的默认值，使用mydict.get(k, <default>)而不是mydict[k]

推荐阅读