itertools 中的 takewhile 取一个值太多

Question

我有以下代码在某个值上停止迭代器，将状态保存到该值并返回保存的状态和原始状态。我正在使用 itertools 中的 takewhile 来获取给定 break_point 之前的值，然后在保存的迭代器上使用 chain 直到 break_point 和初始迭代器来合并它们：

from itertools import takewhile, chain

def iter_break(iterator_input, break_point):

    new_iter = list(takewhile(lambda x: x <= break_point-1, iterator_input))
    return chain(iter(new_iter), iterator_input)

import unittest


class TestEqual(unittest.TestCase):

    def test_iters(self):

        it = iter(range(20))
        old_it = iter_break(it, 10)
        self.assertEqual(list(it), list(old_it))

if __name__ == '__main__':
    unittest.main()

问题是，最终返回的迭代器和我返回的完整迭代器并不相似，因为返回的迭代器错过了一个值，它错过了等于断点本身的值。请帮忙。

Answer 1

it 不仅缺少断点值，它还缺少之前的所有值，因为它只是一个用iter 创建的迭代器，而不是一个列表，所以iter_sample 在用@ 迭代它们时使用了值987654324@。这包括断点本身，因为takewhile 需要查看该值才能知道条件不再满足。