import itertools

def cerca_suma1(it):
    s = 0
    for x in it:
        if x == s:
            yield x
        s = s + x
        
def cerca_suma2(it):
    it1, it2 = itertools.tee(it)
    it3 = itertools.accumulate(it2, initial=0)
    it4 = filter(lambda t: t[0]==t[1], zip(it1, it3))
    it5 = map(lambda t:t[0], it4)
    return it5

# Tria la versió que vols provar
cerca_suma = cerca_suma1