为什么 itertools.dropwhile 只能对列表起作用一次？

Question

我无法理解 itertools.dropwhile() 的以下行为。比方说，我有一个清单：numbers = [1,2,3,4,5]我这样重写：numbers = list(itertools.dropwhile(lambda x: sum(numbers[0:x]) <=7, numbers))值 [1,2,3] 的总和小于 7，因此我预计它们将被删除。事实上，输出是：>>>numbers [4, 5]我重复：numbers = list(itertools.dropwhile(lambda x: sum(numbers[0:x]) <=7, numbers))我预计输出应该是[5]现在。但相反，我再次看到：>>>numbers [4, 5]为什么会发生这种情况？

Answer 1

您的代码没有按照您的想法进行操作。您传递给的可调用函数dropwhile获取元素，而不是索引。

第一次调用结果的原因是[4, 5]：

• sum(numbers[0:1]) <= 7

• sum(numbers[0:2]) <= 7

• sum(numbers[0:3]) <= 7

都小于或等于 7。

在第二次调用中， 和numbers[0:4]都是numbers[0:5]整个列表，其总和始终为 9。

---

您可以dropwhile通过枚举来代替：

>>> numbers = [1, 2, 3, 4, 5]

>>> numbers = [element for i, element in itertools.dropwhile(lambda x: sum(numbers[0:x[0]+1]) <= 7, enumerate(numbers))]

>>> numbers

[4, 5]

>>> numbers = [element for i, element in itertools.dropwhile(lambda x: sum(numbers[0:x[0]+1]) <= 7, enumerate(numbers))]

>>> numbers

[5]

>>> numbers = [element for i, element in itertools.dropwhile(lambda x: sum(numbers[0:x[0]+1]) <= 7, enumerate(numbers))]

>>> numbers

[]