Używanie itertools do aplikacji funkcji rekursywnej

Question

Potrzebuję funkcji Python iterate(f, x), która tworzy iterator zwracający wartości x, f (x), f (f (x)), f (f (f (x))) itp (jak np. Clojure's iterate). Przede wszystkim zastanawiałem się: czy to już istnieje w standardowej bibliotece i brakuje mi jej tylko? Oczywiście jest to dość łatwe do wdrożenia z generatorem:

def iterate(f, x): 
    while True: 
     yield x 
     x = f(x) 

Tak z ciekawości: Czy jest bardziej funkcjonalny sposób to zrobić w Pythonie, np z niektórymi magicznymi narzędziami lub magicznymi funkcjami?

W Pythonie 3.3 to będzie działać

def iterate(f, x): 
    return accumulate(repeat(x), lambda acc, _ : f(acc)) 

ale wygląda nadużycia do mnie. Czy mogę to zrobić ładniej?

Answer 1

Wydaje się, że nie ma czegoś w narzędziach, które robią to, co chcesz, ale itertools to głęboka skrzynia skarbów, więc mogłem coś przeoczyć.

Twój kod generatora wygląda świetnie. Nie wiem, dlaczego napisałbyś to z akumulatorką, chyba że grasz w absurdalną grę w golfa kodu lub próbujesz zrobić wrażenie na snobach Haskella. Napisz swoją funkcję, aby była czytelna, zrozumiała i możliwa do utrzymania. Nie trzeba być zbyt sprytnym.

Answer 2

można użyć anamorphism (lub rozwijać) uproszczenie definicji iterate i używać tylko jedną wartość wyjściową. Poniżej znajduje się realizacja I kiedyś, w oparciu o dość znanym paper:

def ana(build, predicate): 
    def h(x): 
     if predicate(x): 
      return 
     else: 
      a, b = build(x) 
      yield a 
      for i in h(b): 
       yield i 
      # with newer syntax: 
      # yield from h(b) 
    return h 

Wdrażanie iterate z ana wówczas wygląda następująco:

def iterate(f, x): 
    return ana(lambda x: (x, f(x)), lambda _: False)(x) 

Brak itertools, chociaż ... I zgadzam się, że nie jest to najbardziej czytelny wariant. W rzeczywistości jest to raczej zagadkowe.

---

AKTUALIZACJA: Istnieje łatwiejsza wersja, która wygląda nawet ładnie. Zajęło z here:

def unfold(f, x): 
    while True: 
     w, x = f(x) 
     yield w 

I to daje:

def iterate(f, x): 
    return unfold(lambda y: (y, f(y)), x)