在 Python 中循环列表并删除条目

Question

我想在 Python 中遍历一个列表并删除特定项目。 我不想制作一个新的接受项目列表，因为在我的完整示例中，我想对我的列表进行一系列改进。这是一个简单的示例，我尝试删除列表中所有小于 3 的数字。

example = [1.,2.,3.,4.,5.,6.]
for e in example:
  if e < 3.:
    print "Removing:", e
    example.remove(e)
  else:
    print "Accepting:", e
print "NAIVE:", example 

Removing: 1.0
Accepting: 3.0
Accepting: 4.0
Accepting: 5.0
Accepting: 6.0
NAIVE: [2.0, 3.0, 4.0, 5.0, 6.0]

失败了。我认为它失败了，因为删除列表中的项目与 for 循环运行的索引混淆，即一旦项目 1. 被删除，项目 2. 在列表中的位置 0，但是那时，循环在位置 1。

我可以使用deepcopy 解决此问题，如下所示：

example = [1.,2.,3.,4.,5.,6.]
import copy
for e in copy.deepcopy(example):
  if e < 3.:
    print "Removing:", e
    example.remove(e)
  else:
    print "Accepting:", e
print "DEEPCOPY:",  example

Removing: 1.0
Removing: 2.0
Accepting: 3.0
Accepting: 4.0
Accepting: 5.0
Accepting: 6.0
DEEPCOPY: [3.0, 4.0, 5.0, 6.0]

这在这里有效，但这是一种好的做法吗？会导致其他意想不到的错误吗？有没有更好的方法来实现这一目标？还是这种结构（循环并从列表中删除）根本不合理？

我不想创建新的已接受项目列表，因为我想将一系列标准逐个应用于我的列表，并相应地删除项目。我不想为我应用的每个标准（可能很多）创建一个新列表，我也不想一次性应用我的所有标准（因为查看每个标准删除了多少项目等很有帮助） .

Answer 1

我不明白你为什么不只是用你想要保留的项目构建一个新列表，因为你似乎并不关心构建一个新列表（毕竟，这就是 copy 所做的)。

所以我会这样做

example = [f for f in example if f >= 3]

如果您确实想遍历列表并更改它，也许可以遍历索引然后倒退：

for i in range(len(example) - 1, -1, -1):
    if example[i] < 3:
        del example[i]

但这有点特殊，除非真的必要，否则我会避免它。

为了表明您不需要愚蠢的 example_1、example_2、old_example 等变量，请考虑：

# Here is a number of tests for things we want throw out
def some_really_complicated_test_on_a_number(f):
    ... put any kind of code here and return True if we want to
    delete the number...

TESTS = (
    lambda f: f < 3,
    lambda f: f > 16,
    lambda f: (int(f) % 2) == 1,  # Integer is odd
    some_really_complicated_test_on_a_number,
    # etc
)

这是一个函数，它接受一个列表和一个测试，打印带有“接受”和“拒绝”的项目，并返回一个包含剩余项目的新列表：

def filter_with_prints(l, test):
    result = []
    for f in l:
         if test(f):
             print("Rejecting: {}".format(f))
         else:
             result.append(f)
             print("Accepting: {}".format(f))
    return result

我们可以这样调用很多测试：

example = [1., 2., 3., 4., 5., 6.]

for test in TESTS:
    example = filter_with_prints(example, test)

Answer 2

你是对的，问题是你正在修改你在循环期间迭代的列表。这是非常不一致的，并导致许多错误。我的问题是为什么您对删除列表中的项目特别感兴趣，而不是生成符合您建议的新副本？对此有具体要求吗？否则，我建议制作符合您限制的列表的新副本，而不是修改输入列表。所以，修改你的代码：

example = [1.,2.,3.,4.,5.,6.]
new_list = []
for e in example:
   if e >= 3.:
      new_list.append(e)
      print "Accepting:", e
   else:
      print "Removing: ", e

这不太容易出错，但您可以更 Python 并为此使用列表推导：

new_list = [e for e in example if e >= 3.]

编辑：我知道您想要删除项目而不是创建新列表的原因是您要多次浏览列表以过滤列表。我仍然认为即使在这种情况下，每次创建新列表的可读性更高、更不容易出错并且效率也不会特别低。如果效率是问题并且您有非常大的列表或类似的东西，我会尝试只遍历列表一次并在同一个循环中删除所有无效项目。但是，如果您真的想要从列表中删除项目，您可以按照@RemcoGerlich 的说明进行操作，然后按索引向后迭代。