Python中list循环遍历删除数据的正确方法

Python列表遍历删除元素的常见错误

前言

在Python学习初期，常遇到一个典型问题：在遍历列表（list）的同时删除满足特定条件的元素，此时程序往往会抛出异常。示例代码如下：

num_list = [1, 2, 3, 4, 5]
print(num_list)

for i in range(len(num_list)):
 if num_list[i] == 2:
  num_list.pop(i)
 else:
  print(num_list[i])

print(num_list)

运行上述代码会触发以下异常：IndexError: list index out of range

出现该异常的根本原因在于，当从list中删除元素后，列表的实际长度会减小，但for循环预设的遍历次数仍基于原始长度，导致后续访问时索引超出当前列表范围，从而引发溢出错误。

针对此问题，尝试修改代码逻辑如下：

num_list = [1, 2, 3, 4, 5]
print(num_list)

for i in range(len(num_list)):
 if i >= len(num_list):
  break

 if num_list[i] == 2:
  num_list.pop(i)
 else:
  print(num_list[i])

print(num_list)

此时程序不再抛出异常，但打印输出结果如下：

[1, 2, 3, 4, 5] 1 4 5 [1, 3, 4, 5] [Finished in 0.441s]

虽然最终列表中的元素[2]确实被成功删除，但循环过程中的打印结果存在偏差，遗漏了元素[3]的输出。

经分析可知原因：当满足条件删除元素[2]后，后续元素整体向前移动一位，即[3, 4, 5]前移，此时原[3]的索引变为1（原[2]的位置）。然而，下一次for循环迭代时，索引从2开始取值，直接获取到的是[4]，从而跳过了[3]。

若将代码调整为以下形式，结果依然相同，问题并未解决：

num_list = [1, 2, 3, 4, 5]
print(num_list)

for item in num_list:
 if item == 2:
  num_list.remove(item)
 else:
  print(item)

print(num_list)

既然已明确问题根源，寻找正确解决方案便不再困难。以下是修正后的代码：

num_list = [1, 2, 3, 4, 5]
print(num_list)

i = 0
while i < len(num_list):
 if num_list[i] == 2:
  num_list.pop(i)
  i -= 1
 else:
  print(num_list[i])

 i += 1

print(num_list)

执行该代码后，结果完全符合预期：

[1, 2, 3, 4, 5] 1 3 4 5 [1, 3, 4, 5] [Finished in 0.536s]

使用while循环手动调整索引

我的思路是：既然for循环存在缺陷，便改用while循环实现。每次while循环都会重新检查list的长度（通过判断i < len(num_list)），从而避免索引溢出。在删除元素[2]后，手动将当前索引减1（i -= 1），确保下一次循环通过调整后的索引能正确获取元素[3]，而非跳过它。

通用解决方案：倒序遍历与副本操作

当然，上述方法并非最优解。经过搜索，发现两种通用的标准解决方案：1. 倒序循环遍历；2. 遍历列表的拷贝，操作原始列表。

1、倒序循环遍历：

num_list = [1, 2, 3, 4, 5]
print(num_list)

for i in range(len(num_list)-1, -1, -1):
 if num_list[i] == 2:
  num_list.pop(i)
 else:
  print(num_list[i])

print(num_list)

执行结果完全正确。为何正序删除会出错，而倒序删除却可行？此处难以用文字完全阐述，建议结合图示理解。

1）正序循环时删除：

删除元素[2]后，下一次循环索引为2，此时该位置存放的是[4]，导致[3]被遗漏。

2）倒序循环时删除：

删除元素[2]后，[3, 4, 5]向前移动，但由于倒序循环下一次索引为0（或递减方向），该位置存放的是[1]，这正是我们期望的正确元素值，因此不会遗漏。

2、遍历拷贝的list，操作原始的list：

num_list = [1, 2, 3, 4, 5]
print(num_list)

for item in num_list[:]:
 if item == 2:
  num_list.remove(item)
 else:
  print(item)

print(num_list)

若原始列表为num_list，则num_list[:]生成了其浅拷贝，是一个新的独立列表。通过遍历这个新列表，同时操作原始列表进行删除，既不会引发索引溢出，也能得到期望的最终结果。此方法的潜在缺点在于，若列表过大，拷贝操作可能占用较多内存。针对此类情况，推荐使用倒序遍历方法以节省内存。

总结

总结

以上就是本文的全部内容，希望这些内容对大家的学习或工作具有一定的参考价值和指导意义。感谢大家对本站的支持。