为什么 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]) <= 7sum(numbers[0:2]) <= 7sum(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[]