Rückgabeliste aus einer Schleife mit Lambda

Question

Ich habe eine Funktion, um eine Liste von Dateien und Ordnern in einem bestimmten Ordner (mit recurse und nur Dateien erhalten Optionen) oder nur die Datei in einer Liste, wenn der angegebene Pfad ist nicht Ordner:

def path_to_list(path, onlyFiles = False, recurse = False): 
    if os.path.isdir(path): 
     if onlyFiles: 
      if recurse: 
       result = [] 
       for dirs in list(os.walk(path)): 
        result.append(dirs[2]) 
       return result 
      else: 
       return next(os.walk(path))[2] 
    else: 
     return list(os.walk(path)) if recurse else next(os.walk(path)) 

    return [path] 

Der Versuch, diesen Teil zu verkürzen:

result = [] 
for dirs in list(os.walk(path)): 
    result.append(dirs[2]) 
return result 

ich versuchte Lambda mit mehreren Syntaxen zu verwenden, haben aber keinen Erfolg bekam. Wie gebe ich das Ergebnis direkt aus der for-Schleife zurück? Vielen Dank.

Answer 1

Sie können Verzeichnisliste zurückkehren, indem Sie:

return [dir for _, _, dir in list(os.walk(path))]