Python3中for循环踩过的坑记录

问题背景

在Python3中，for循环是常用的一种语句结构，但在使用过程中，我们也会遇到一些问题和坑，下面我将分享一些我在使用for循环时遇到过的问题。

问题1：循环中修改列表导致问题

当我们在for循环中对列表进行修改时，经常会出现一些奇怪的问题，例如：

lst = [1, 2, 3, 4, 5]
for i in lst:
    if i == 2:
        lst.remove(i)
print(lst)

上述代码的期望输出为[1, 3, 4, 5]，但实际输出为[1, 3, 4, 5]，也就是说并没有将2从列表中删除。

这是因为在for循环中，通过lst[i]访问列表，其实是通过索引来访问列表的，当我们删除一个元素后，列表的长度就会发生变化，这会导致索引出现问题，进而修改列表时出现错误。

解决这个问题的方法是用一个新的列表来保存需要删除的元素，然后再遍历这个列表进行删除操作，例如：

lst = [1, 2, 3, 4, 5]
to_remove = []
for i in lst:
    if i == 2:
        to_remove.append(i)
for i in to_remove:
    lst.remove(i)
print(lst)

问题2：循环遍历字典时要慎用dict.items()

在Python3中，循环遍历字典可以使用dict.keys()、dict.values()和dict.items()，其中dict.items()可以同时返回字典中的键值对，这使得在某些场景下使用dict.items()更加方便。

但是，我们在for循环遍历字典时，应当尽量避免使用dict.items()，因为它会创建一个临时的列表，将所有的键值对都存放在这个列表中，然后再对这个列表进行遍历操作，这会导致性能问题。

例如：

d = {'a': 1, 'b': 2, 'c': 3}
for k, v in d.items():
    print(k, v)

上述代码的期望输出为：

a 1
b 2
c 3

但其实在运行过程中，Python会创建一个临时的列表，将所有的键值对存放在这个列表中，然后再遍历这个列表，这会导致性能问题。

解决这个问题的方法是使用dict.keys()或dict.values()，例如：

d = {'a': 1, 'b': 2, 'c': 3}
for k in d.keys():
    print(k, d[k])

或者：

d = {'a': 1, 'b': 2, 'c': 3}
for v in d.values():
    print(v)

如果需要同时访问键和值，也可以使用zip()函数：

d = {'a': 1, 'b': 2, 'c': 3}
for k, v in zip(d.keys(), d.values()):
    print(k, v)

总结

本文介绍了Python3中for循环中的两个常见问题，并分别给出了相应的解决方法。当我们在编写代码时，应该尽量遵循Python的语法规范，这样可以避免一些常见的错误和坑。