如何用insert方法在Python列表指定元素前动态插入新元素

给定原始列表：

list1 = ['a', 'b', 'c', 'd', 'e', 'f', 'g']

以及目标元素列表：

list2 = ['a', 'd', 'e']

需要在list1中每个属于list2的元素前插入{char}1格式的元素，最终得到：

['a1', 'a', 'b', 'c', 'd1', 'd', 'e1', 'e', 'f', 'g']

为什么正序插入会失效？ #

直接正序遍历索引并执行insert时，每次插入都会让后续元素的索引+1，导致之前记录的索引位置不再对应原来的元素，最终只有第一个插入操作能正确生效。

用insert实现的高效方法：倒序处理索引 #

核心思路是从后往前处理需要插入的位置，这样后续插入操作不会影响尚未处理的元素的原始索引。具体步骤：

1. 将list2转为集合，提升元素存在性检查的效率（集合的in操作是O(1)，列表是O(n)）。
2. 收集所有需要插入元素的原始索引。
3. 倒序遍历这些索引，执行插入操作。

代码实现：

list1 = ['a', 'b', 'c', 'd', 'e', 'f', 'g']
list2 = ['a', 'd', 'e']

# 转集合优化查询
target_chars = set(list2)
# 获取所有目标元素的原始索引
insert_indices = [idx for idx, char in enumerate(list1) if char in target_chars]

# 倒序遍历索引，避免插入导致的索引偏移
for idx in reversed(insert_indices):
    list1.insert(idx, f"{list1[idx]}1")

print(list1)

更高效的替代方案：生成新列表 #

如果允许创建新列表而非修改原列表，这种方法时间复杂度更低（O(n)），无需处理索引偏移问题：

list1 = ['a', 'b', 'c', 'd', 'e', 'f', 'g']
list2 = set(['a', 'd', 'e'])

result = []
for char in list1:
    if char in list2:
        result.append(f"{char}1")
    result.append(char)

print(result)

内容的提问来源于stack exchange，提问作者JJ Kam