如果满足条件，如何在遍历列表时在列表中插入值？

Question

问题; 遍历这个列表，无论你在哪里找到一个 2，修改列表，这样你就可以在它的前后加上标志“Found”和“Target”。预期输入：l = [1,0,1,0,2,1,0,2,0,1]预期输出：l = [1,0,1,0,"Found",2,"Target", 1,0,"找到",2,"目标",0,1]看起来很容易？l = [1,0,1,0,2,1,0,2,0,1]for i in range(len(l)):    if l[i] == 2:        l.insert(i-1,"Found")        l.insert(i-+,"Target")但这会失败，但说明了这个想法？我已经搜索了很多堆栈问题，但我似乎无法找到一个优雅的解决方案。我觉得我也在用错误的逻辑来解决这个问题，因为我相信你不应该改变你正在迭代的列表？

Answer 1

是的，你是对的......你不应该改变你正在迭代的列表。因此，您需要创建列表的两个版本...一个是原始版本 ( l)，另一个是迭代后的原始版本 ( new_l)。

这就是我的意思：

l = [1,0,1,0,2,1,0,2,0,1]

new_l = []

for i in range(len(l)):

    if l[i] == 2:

        new_l.extend(["Found", l[i], "Target"])

    else:

        new_l.append(l[i])

print(new_l)

# [1,0,1,0,"Found",2,"Target",1,0,"Found",2,"Target",0,1]

Answer 2

在迭代对象时更改对象通常不是一个好主意。

这就是我将如何做到这一点

l = [1,0,1,0,2,1,0,2,0,1]

new_l = []

for el in l:

    if el == 2:

        new_l.extend(["Found", el, "Target"])

    else:

        new_l.append(el)

请注意，我使用的是更“pythonic”且更高效的方法来遍历列表，而不是使用 range(len(l))。

第二种方法，我不会特别使用，但它可以给你另一个想法是使用 list comprhension + itertools.chain.from_iterable（itertools 是一个标准库包）。

new_l = (["Found", el, "Target"] if el == 2 else [el] for el in l)

new_l = list(itertools.chain.from_iterable(new_l))

Answer 3

如果我们不想创建新列表而只想修改第一个。

这个使用与您相同的逻辑，但是我们必须使用全局索引进行迭代，这样我们就不会在重复的位置追加，也可以跳过不需要的位置。

l = [1,0,1,0,2,1,0,2,0,1,2]

print(l)

k = 0

for i in range(len(l)):

    if l[k] == 2:

        l.insert(k,"Found")

        l.insert(k+2,"Target")

        k+=3

    else:

        k += 1

print(l)

[1, 0, 1, 0, 2, 1, 0, 2, 0, 1, 2]

[1, 0, 1, 0, 'Found', 2, 'Target', 1, 0, 'Found', 2, 'Target', 0, 1, 'Found', 2, 'Target']