具有特定扩展名的文件的递归搜索

Question

我正在试图获取以特定扩展名（哪个用户将通过）与路径结尾的所有文件的计数。我们也有子文件夹，因此搜索必须递归。 下面是我正在尝试，但它是抛出错误。请建议差距在哪里。 如果删除if file.endswith(extension):线然后它给出的所有文件的计数值（它包括与所有扩展名的文件）

import os, sys 

def fileCount(path, extension): 
    count = 0 
    for root, dirs, file in os.walk(path): 
     if file.endswith(extension): 
      count += len(file) 
    return count 

print fileCount('/home/export/JobDefinition', '.car') 

下面是输出：通过文件小号

$ python test.py 
Traceback (most recent call last): 
    File "test.py", line 11, in <module> 
    print fileCount('/home/export/JobDefinition', '.car') 
    File "test.py", line 6, in fileCount 
    if file.endswith(extension): 
AttributeError: 'list' object has no attribute 'endswith' 

Answer 1

你想的和过滤的所有文件后：

def fileCount(path, extension): 
    count = 0 
    for root, dirs, files in os.walk(path): 
     count += sum(f.endswith(extension) for f in files) 
    return count 

files返回文件的列表，以便sum(f.endswith(extension) for f in files)会给你指定的扩展名结尾的所有文件的计数。

或者只是返回所有的总和：

def fileCount(path, extension): 
    return sum(f.endswith(extension) for root, dirs, files in os.walk(path) for f in files) 

Answer 2

os.walk()用于迭代。

您必须遍历文件S，它们以list的形式返回。

def fileCount(path, extension): 
    count = 0 
    for root, dirs, files in os.walk(path): 
     for file in files: 
      if file.endswith(extension): 
       count += 1 
    return count 

Answer 3

os.walk()返回一个元组，如 - (dirpath, dirnames, filenames)。其中filenames是目录中所有文件的列表，其a list。

您需要遍历文件，而不是使用file.endswith

示例 -

import os, sys 

def fileCount(path, extension): 
    count = 0 
    for root, dirs, files in os.walk(path): 
     for file in files: 
      if file.endswith(extension): 
       count += 1 
    return count 

print fileCount('/home/export/JobDefinition', '.car') 

Answer 4

的os.walk第三返回值是文件名的列表。

Answer 5

这里是一个内衬替代：

import os 

def fileCount(path, extension): 
    return sum([1 for root, dirs, files in os.walk(path) for file in files if file.endswith(extension)]) 

print fileCount('/home/export/JobDefinition', '.car')