python - 在python中动态更改迭代器是否安全？

Question

使用python，我需要在迭代列表时动态地在列表中添加成员：

i = [1,2,3,4,5]

for a in i:
    if a == 1:
        i.append(100)
    print a

能保证正常工作吗？

Answer 1

您正在更改序列，而不是迭代器。for循环调用iter()您正在循环的对象，创建一个循环迭代器。对于一个list对象，创建的列表迭代器保持对原始列表的引用和一个内部索引，并且每次for循环请求下一个项目（next()在迭代器上调用）时，返回当前索引处的项目并增加索引.

在这种情况下，它是安全的，因为循环迭代器对象将愉快地继续处理额外的项目。只要您不i无限期地添加项目，循环就会完成。

但是，如果您要从中删除项目i，则循环迭代器将看不到这些更改，并且您会得到“令人惊讶”的结果：

i = [1, 2, 3, 4, 5]
for a in i:
   if a == 1:
       del i[0]
   print a

印刷：

1
3
4
5

因为循环迭代器i[1]在下一次迭代时产生，即 now 3， not 2。

如果您要在当前迭代索引之前插入项目，也会出现类似的问题。

您可以自己创建迭代器，并手动进行试验以查看会发生什么：

>>> i = [1, 2, 3, 4, 5]
>>> i_iter = iter(i)  # index starts at 0
>>> next(i_iter)      # now the index is 1
1
>>> del i[0]
>>> i
[2, 3, 4, 5]          # item at index 1 is 3
>>> next(i_iter)      # returns 3, next index is 2
3
>>> i.insert(0, 1)
>>> i
[1, 2, 3, 4, 5]       # item at index 2 is 3
>>> next(i_iter)      # returns 3, next index is 3
3