Wydajność w funkcji rekurencyjnej

Iteratory nie działają rekurencyjnie w ten sposób. Musisz ponownie uzyskać każdy wynik, zastępując

explore(path)

Z czymś w rodzaju

for value in explore(path):
    yield value

Python 3.3 dodał składnię yield from X, jak zaproponowano w PEP 380 , aby służyć temu celowi. Z nim możesz zrobić to zamiast:

yield from explore(path)

Jeśli używasz generatorów jako koroutines , ta składnia obsługuje również użycie generator.send() przekazywanie wartości z powrotem do wywoływanych rekurencyjnie generatorów. The simple for pętla powyżej nie będzie.

Problemem jest ta linijka kodu:

explore(path)

Co to robi?

• połączenia explore z nowym path
• explore działa, tworząc generator
• generator jest zwracany do miejsca, w którym explore(path) został wykonany . . .
• i jest odrzucany

Dlaczego jest odrzucony? Nie był do niczego przypisany , nie był iterowany, został całkowicie zignorowany.

Jeśli chcesz coś zrobić z wynikami, cóż, masz żeby coś z nimi zrobić! ;)

Najprostszym sposobem na poprawienie kodu jest:

for name in explore(path):
    yield name

Kiedy jesteś pewien, że rozumiesz, co się dzieje, prawdopodobnie będziesz chciał użyć os.walk() zamiast tego.

Po migracji do Pythona 3.3 (zakładając, że wszystko działa zgodnie z planem) będziesz mógł użyć nowej składni yield from i najprostszym sposobem na poprawienie kodu w tym momencie będzie:

yield from explore(path)

Zmień to:

explore(path)

Do tego:

for subpath in explore(path):
    yield subpath

Lub użyj os.walk, Jak zasugerował phooji (co jest lepszą opcją).

Wywołujący {[1] } Jak funkcję. To, co powinieneś zrobić, to iterację go jak generatora:

if stat.S_ISDIR(stat_info.st_mode):
  for p in explore(path):
    yield p
else:
  yield path

EDIT: zamiast modułu stat możesz użyć os.path.isdir(path).

Spróbuj tego:

if stat.S_ISDIR(stat_info.st_mode):
    for p in explore(path):
        yield p

Os.spacer jest świetny, jeśli chcesz przejść wszystkie foldery i podfoldery. Jeśli tego nie potrzebujesz, to jak użycie pistoletu na słonie do zabicia muchy.

Jednak w tym konkretnym przypadku, os.spacer może być lepszym podejściem.

Można również zaimplementować rekurencję za pomocą stosu.

import os
import stat

def explore(p):
    '''
    perform a depth first search and yield the path elements in dfs order
        -implement the recursion using a stack because a python can't yield within a nested function call
    '''
    list_t=type(list())
    st=[[p,0]]
    while len(st)>0:
        x=st[-1][0]
        print x
        i=st[-1][1]

        if type(x)==list_t:
            if i>=len(x):
                st.pop(-1)
            else:
                st[-1][1]+=1
                st.append([x[i],0])
        else:
            st.pop(-1)
            stat_info = os.lstat(x)
            if stat.S_ISDIR(stat_info.st_mode):
                st.append([['%s/%s'%(x,a) for a in os.listdir(x)],0])
            else:
                yield x

print list(explore('.'))