具有特定擴展名的文件的遞歸搜索

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')