python - 如何使用追加对列表中的每个项目进行平方？

Question

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

for i in start_list:
    start_list.append(i**2)

我假设在迭代发生时每个列表项的值都分配给 i。

但考虑到上面的代码永远不会完成 - 看起来列表正在被无限追加。

为什么会这样？

我怎样才能正确地编写它，以便它使用追加函数对每个元素进行平方。

Answer 1

for 循环仅在到达列表对象的最后一个元素时停止；通过在循环体中添加一个新元素，列表中总是会有更多元素。

在迭代时使用列表的副本，使用索引，或使用带有列表理解的 list.extend():

for i in start_list[:]:  # a copy won't grow anymore.
    start_list.append(i ** 2)

或

for idx in range(len(start_list)):  # the range is created just once
    start_list.append(start_list[idx] ** 2)

或

start_list.extend([i ** 2 for i in start_list]) # Append items from a new list
start_list += [i ** 2 for i in start_list]  # ditto, different spelling

如果您确实打算用它们的平方等值替换这些值，只需从旧列表生成一个新列表:

start_list = [i ** 2 for i in start_list]