python zadanie z użyciem funkcji generatora

Question

Cześć ! Mam do napisania funkcję generatora która przyjmuje takie parametry (sekwencja,*sekwencje). Akceptowane argumenty to dowolne sekwencje . Jeżeli do funkcji przekazano jedynie sekwencję to zwracany generator zawiera elementy tej sekwencji cyklicznie powtórzone nieskończoną ilość razy . Jeżeli sekwencji jest więcej niż jedna to zwracany generator zawiera krotki elementów równolegle pobrane z sekwencji, przy czym każda jest cykliczne powtórzona nieskończenie wiele razy (xyz, ac => (x,a), (y.b),(z,a) ...) Jeżeli którykolwiek z argumentów jest pusty to zwracany generator też jest pusty.

from itertools import permutations, cycle 
def cykliczny(sekwencja,*sekwencje):
    wynik = []
    if sekwencje:
        for s1 in sekwencja:
            for s2 in sekwencje:
                wynik.extend([list(zip(s1,p))for p in permutations(s2)])
            return wynik
        else:
            return ''.join(cycle(sekwencja))

Dopiero od niedawna uczę się python dlatego proszę o wyrozumiałość . :) 

Czy macie jakieś sugestie co mogę poprawić żeby działało poprawnie? 

Answer 1

Trochę nie tędy droga. W języku Python można na kilka sposobów napisać generator:

1. Zwracać z funkcji dane z użyciem yield (będzie wtedy działać domknięcie)

2. Użyć postaci generatorowej i otoczyć ją nawiasami zwykłymi (w wyniku powstanie generator)

3. Użyć wersji obiektowej implementacji generatora (trzeba tworzyć wtedy klasę i zaimplementować odpowiednie metody).

W tym zadaniu masz 2 ścieżki wykonania i to zrobione jest dobrze. W jednej zwracasz proste itertools.cycle(sekwencja) a w drugiej, zip(...) na "opakowanych itertools.cycle(...) sekwencjach".

No to błędny jeszcze kod:

#!/usr/bin/env python3

def cykliczny(sekwencja, *sekwencje):
    from itertools import cycle
    if not sekwencje:
        return cycle(sekwencja)
    return zip(cycle(sek) for sek in (sekwencja, *sekwencje))

if __name__ == '__main__':
    # Dwa testy, pierwsze 10 elementów... 
    for k, v in zip(range(10), cykliczny('abcd')):
        print(v)

    for k, v in zip(range(10), cykliczny('abcd', 'gh', 'qwer')):
        print(v)

Pierwsza ścieżka jest już ok. W drugiej niestety zwraca 3 iteratory:

a
b
c
d
a
b
c
d
a
b
(<itertools.cycle object at 0x1095f3798>,)
(<itertools.cycle object at 0x1095f3750>,)
(<itertools.cycle object at 0x1095f37e0>,)

Czyli trzeba przerobić argumenty zip'a tak aby zwracał rozwinięte argumenty. Chodzi o ten kod

cycle(sek) for sek in (sekwencja, *sekwencje)

Wystarczy że opakujesz je nawiasami ( i ) i będzie z tego generator. Ale niestety nie zostanie on rozwinięty dla zip'a. Wystarczy dodać * (gwiazdkę), co spowoduje rozwinięcie argumentów:

zip(*(cycle(sek) for sek in (sekwencja, *sekwencje)))

Sprawdź jak to działa :)

Jak zauważysz w kodzie, zwracasz albo cykl z 1 sekwencji, albo "zzipowane cykle z pierwszej i dalszych sekwencji". Do takich wyrażeń jest w języku użycie "if po" :) 

#!/usr/bin/env python3

def cykliczny(sekwencja, *sekwencje):
    from itertools import cycle
    return cycle(sekwencja) if not sekwencje else zip(*(cycle(sek) for sek in (sekwencja, *sekwencje)))

if __name__ == '__main__':
    # Dwa testy...
    for k, v in zip(range(10), cykliczny('abcd')):
        print(v)

    for k, v in zip(range(10), cykliczny('abcd', 'gh', 'qwer')):
        print(v)

Teraz tylko wystarczy zauważyć że tak naprawdę wystarczy w implementacji jedna ścieżka wykonania. To skończy implementację:

#!/usr/bin/env python3

def cykliczny(*args):
    from itertools import cycle
    return zip(*(cycle(sek) for sek in args))

if __name__ == '__main__':
    # Dwa testy...
    for k, v in zip(range(10), cykliczny('abcd')):
        print(v)
    for k, v in zip(range(10), cykliczny('abcd', 'gh', 'qwer')):
        print(v)

 

Comments

Zaspoilowałeś rozwiązanie, miałem nadzieję że powoli do tego sam dojdzie. Mógłbyś proszę schować rozwiązania/odpowiedź proszę? :)

Answer 2

Przede wszystkim, to nie generator. Kluczowym komponentem generatora jest słowo kluczowe 'yield', którego tutaj nie widzę. (zwracania generatora przez `return cycle()` technicznie wygląda jak generator, ale nie jestem przekonany że o to w tym zadaniu chodziło)

return ''.join(cycle(sekwencja))

To jest efektywnie nieskończona pętla - ''.join() nigdy nie wyjdzie. Nie rozumiem, czemu w ogóle tu są operacje na stringach.

else:

to raczej powinno być `else` do `if`a, nie do pętli.

wynik.extend([list(zip(s1,p))for p in permutations(s2)])

Pomijając fakt, że to znowu nie wygląda jak generator, to też nie wygląda jak to, czego wymaga treść zadania. Z czytania treści, nie wydaje mi się by permutations() było potrzebne - jedynie połączenie cycle() i zip().

Comments

Zestawienie for/else występuje w Pythonie.

---

Tak, wiem. Ale patrząc na treść zadania, jestem mocno przekonany że w tym przypadku `else` zostało umieszczone w złym miejscu

---

Rzeczywiście, ach te wcięcia

---

Faktycznie , nie zauważyłam wcięcia.  Poprawiłam tamto i zastąpiłam cycle iteratorem , ale nadal nie działa 

def cykliczny(sekwencja,*sekwencje):
    wynik = []
    if sekwencje:
        for s1 in sekwencja:
            for s2 in sekwencje:
                wynik.extend([list(zip(s1,s2))])
            yield wynik
    else:
        for element in sekwencja:
            yield element
            saved.append(element)
        while saved:
            for element in saved:
                  yield element

 

---

Druga część (bez dodatkowych sekwencji) wygląda prawie poprawnie (choć przekombinowanie), bo nigdzie nie stworzyłeś `saved`.

W pierwszej części, przede wszystkim, nie widzę nieskończonej iteracji. Poza tym, `yield`ujesz listę list, co też wygląda dziwnie.

Answer 3

Napisałem taki kodzik na szybko, pewnie dałoby się jakoś ładniej / lepiej ale ważne że działa 
 

def multiple_generator(*args):
    generator_list = []

    def get_next_elem(sequence):
        while True:
            for i in sequence:
                yield i

    if len(args) == 1:
        gen = get_next_elem(*args)
        while True:
            yield next(gen)

    elif len(args) == 0:
        while True:
            yield None

    else:
        for j in args:
            generator_list.append(get_next_elem(j))

        while True:
            yield tuple(next(x) for x in generator_list)

arr = [2, 3, 7, 8]
arr2 = [1, 4, 0, 3]
arr3 = [8, 4, 3, 0]


a = multiple_generator(arr, arr2, arr3)
for i in range(50):
    print(next(a))