在 Python 列表和字典常踩的一个坑

当我们在 Python 中处理数据时，难免会遇到列表和字典。这两个数据结构非常强大，可以帮助我们存储和处理各种类型的数据。然而，在使用它们时，我们也常常会遇到一些坑。今天，我们就来聊聊在 Python 列表和字典中常踩的一个坑。

当我们想要遍历一组数据并对它们进行修改时，通常会使用 for 循环。然而，如果我们直接在 for 循环中修改列表或字典的元素，可能会出现一些意想不到的问题。

假设我们有一个列表如下：

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

我们想将列表中的每个元素都乘以 2。我们可以使用以下代码来实现：

for i in range(len(my_list)):
    my_list[i] *= 2

这样，列表中的每个元素都会被乘以 2。然而，如果我们想在遍历列表的同时，添加新的元素，就会出现问题。

for i in range(len(my_list)):
    my_list[i] *= 2
    my_list.append(i + 6)

上面的代码会报错，因为在遍历列表的同时，我们改变了列表的长度。这会导致索引 i 超出列表的范围。

解决这个问题的方法是使用 enumerate() 函数。enumerate() 函数可以返回一个枚举对象，枚举对象包含每个元素的索引和值。我们可以使用枚举对象来遍历列表，并在遍历过程中修改列表的元素。

for i, item in enumerate(my_list):
    my_list[i] *= 2
    my_list.append(i + 6)

这样，我们就不会出现索引越界的问题了。

类似地，在字典中，如果我们在遍历字典的同时，修改字典的元素，也可能会出现问题。

假设我们有一个字典如下：

my_dict = {'a': 1, 'b': 2, 'c': 3}

我们想将字典中的每个值都乘以 2。我们可以使用以下代码来实现：

for key in my_dict:
    my_dict[key] *= 2

这样，字典中的每个值都会被乘以 2。然而，如果我们想在遍历字典的同时，添加新的键值对，就会出现问题。

for key in my_dict:
    my_dict[key] *= 2
    my_dict['d'] = 4

上面的代码会报错，因为在遍历字典的同时，我们改变了字典的键值对数量。这会导致在遍历字典时出现 KeyError。

解决这个问题的方法是使用 items() 函数。items() 函数可以返回一个字典项对象，字典项对象包含每个键值对。我们可以使用字典项对象来遍历字典，并在遍历过程中修改字典的元素。

for key, value in my_dict.items():
    my_dict[key] *= 2
    my_dict['d'] = 4

这样，我们就不会出现 KeyError 了。