Python在列表中的每个元素上调用递归函数

Question

我正在研究这个称为traverseDir的递归函数，一切都很顺利，直到这里，我不知道如何迭代列表并在每个元素上调用我的traverseDir函数在列表中。非常感谢，如果你能帮助！

path = sys.argv[1] 

def traverseDir(path): 

    allFile = 0 
    someFile = 0 

    if os.path.isfile(path): # base case 
     print(sys.argv[0]) 
     allFile += 1 
     if path.endswith('.some'): 
      someFile += 1 

    else: 
     files = os.listdir(path) 
     return files 

     #if len(files[0]) <= 1: these 2 lines are where I can't figure out 
     # return traverseDir(item) 

Answer 1

内置函数os.walk已经这样做了。但是，为了您的问题，您需要遍历您的files列表。您还需要通过递归将all_files和some_files向下传递，以便它们可以随时累积。您还需要返回all_files和some_files（python可让您将多个值作为元组返回）。然后，您可以为all_files和some_files添加递归返回的值。

def traverseDir(path, all_files=0, some_files=0): 
    # ... your existing code 

    files = os.listdir(path) 
    for f in files: 
     # extend the path 
     full_path = os.path.join(path, f) 

     # unroll the returned values from the recursion 
     rec_all_files, rec_some_files = traverseDir(full_path, all_files, some_files) 
     # accumulate the values 
     all_files += rec_all_files 
     some_files += rec_some_files 

    return all_files, some_files 

这将在列出的每个文件上调用traverseDir()。反过来，该递归将在其列出的每个文件上调用traverseDir()。

Answer 2

os.listdir

import sys 
import os 

path = sys.argv[1] 


def traverseDir(path): 
    allFile = 0 
    someFile = 0 
    if os.path.isfile(path): 
     allFile += 1 
     if path.endswith('.some'): 
      someFile += 1 
    else: 
     for file in os.listdir(path): 
      agAllFile, agSomeFile = traverseDir(os.path.join(path, file)) 
      allFile += agAllFile 
      someFile += agSomeFile 
    return allFile, someFile 

print traverseDir(path) 

os.walk

import sys 
import os 

path = sys.argv[1] 


def traverseDir(path): 
    files = [file for dirFiles in os.walk(path) for file in dirFiles[2]] 
    return len(files), len([file for file in files if file.endswith('.some')]) 

print traverseDir(path)