itertools内のtakewhileが1つの値をとりすぎる

Question

イテレータを特定の値で停止し、その値まで状態を保存し、保存された状態と元の状態の両方を返す次のコードがあります。私は与えられたbreak_pointまでの値を取得するためにitertoolsからtakewhile使用していますが、その後、私はそれらをマージする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() 

問題があります返されたイテレータと返されている完全なイテレータは、返されたものが1つの値を逃し、ブレークポイント自体に等しい値を見逃しているため、似ていません。助けてください。それはtakewhileでそれらの上に反復して

Answer 1

itはちょうどそれがiterとないリストを作成しただけでイテレータだから、それはそれの前にすべての値を欠けて、ブレークポイントの値が欠落していないので、iter_sampleは値を使用しています。これは、条件がもはや満たされていないことを知るために、takewhileがその値を参照する必要があるため、ブレークポイント自体を含みます。