Python：如何查找具有特定擴展名的所有文件？

Question

我正在嘗試使用Python在目錄中找到所有.c文件。

我寫了這個，但它只是返回所有文件 - 不僅僅是.c文件。

import os 
import re 

results = [] 

for folder in gamefolders: 
    for f in os.listdir(folder): 
     if re.search('.c', f): 
      results += [f] 

print results 

我怎樣才能得到.c文件？

Answer 1

嘗試改變內循環，這樣的事情

results += [each for each in os.listdir(folder) if each.endswith('.c')] 

Answer 2

for _,_,filenames in os.walk(folder): 
    for file in filenames: 
     fileExt=os.path.splitext(file)[-1] 
     if fileExt == '.c': 
      results.append(file) 

Answer 3

試試「水珠」：

>>> import glob 
>>> glob.glob('./[0-9].*') 
['./1.gif', './2.txt'] 
>>> glob.glob('*.gif') 
['1.gif', 'card.gif'] 
>>> glob.glob('?.gif') 
['1.gif'] 

Answer 4

import os, re 
cfile = re.compile("^.*?\.c$") 
results = [] 

for name in os.listdir(directory): 
    if cfile.match(name): 
     results.append(name) 

Answer 5

如果您要更換'.c'與'[.]c$'，您正在搜索包含.c作爲名稱的最後兩個字符的文件，而不是全部包含c的文件，其中至少有一個字符。

編輯：另外，比賽f[-2:]與'.c'，這可能是計算量比拉出一個正則表達式匹配便宜。

Answer 6

KISS

# KISS 

import os 

results = [] 

for folder in gamefolders: 
    for f in os.listdir(folder): 
     if f.endswith('.c'): 
      results.append(f) 

print results 

Answer 7

對於另一種選擇，你可以使用fnmatch

import fnmatch 
import os 

results = [] 
for root, dirs, files in os.walk(path) 
    for _file in files: 
     if fnmatch.fnmatch(_file, '*.c'): 
      results.append(os.path.join(root, _file)) 

print results 

或與列表理解：

for root, dirs, files in os.walk(path) 
    [results.append(os.path.join(root, _file))\ 
     for _file in files if \ 
      fnmatch.fnmatch(_file, '*.c')] 

或使用濾光器：

for root, dirs, files in os.walk(path): 
    [results.append(os.path.join(root, _file))\ 
     for _file in fnmatch.filter(files, '*.c')]  

Answer 8

shutil.copytree的實施是在文檔。我對它進行了mfdified，以列出INCLUDE的擴展名。

def my_copytree(src, dst, symlinks=False, *extentions): 
    """ I modified the 2.7 implementation of shutils.copytree 
    to take a list of extentions to INCLUDE, instead of an ignore list. 
    """ 
    names = os.listdir(src) 
    os.makedirs(dst) 
    errors = [] 
    for name in names: 
     srcname = os.path.join(src, name) 
     dstname = os.path.join(dst, name) 
     try: 
      if symlinks and os.path.islink(srcname): 
       linkto = os.readlink(srcname) 
       os.symlink(linkto, dstname) 
      elif os.path.isdir(srcname): 
       my_copytree(srcname, dstname, symlinks, *extentions) 
      else: 
       ext = os.path.splitext(srcname)[1] 
       if not ext in extentions: 
        # skip the file 
        continue 
       copy2(srcname, dstname) 
      # XXX What about devices, sockets etc.? 
     except (IOError, os.error), why: 
      errors.append((srcname, dstname, str(why))) 
     # catch the Error from the recursive copytree so that we can 
     # continue with other files 
     except Error, err: 
      errors.extend(err.args[0]) 
    try: 
     copystat(src, dst) 
    # except WindowsError: # cant copy file access times on Windows 
    #  pass 
    except OSError, why: 
     errors.extend((src, dst, str(why))) 
    if errors: 
     raise Error(errors) 

用途：例如，僅複製的.config和.BAT文件....

my_copytree（源，TARG， '的.config'， '.BAT' ）

Answer 9

只是要清楚，如果你想在你的搜索詞點字符，你可能已經逃脫它太：

會給你你需要的東西，再加上你需要使用類似「* [反斜線] .C。」：

結果。附加（f），而不是你已經列出的結果+ = [f]

Answer 10

有一個更好的解決方案，直接使用正則表達式，它是用於處理文件名模式的標準庫模塊fnmatch。 （參見glob模塊。）

寫一個輔助函數：

import fnmatch 
import os 

def listdir(dirname, pattern="*"): 
    return fnmatch.filter(os.listdir(dirname), pattern) 

，並用它如下：

result = listdir("./sources", "*.c") 

Answer 11

這個函數返回所有文件名的列表，指定的擴展住在指定目錄：

import os 

def listFiles(path, extension): 
    return [f for f in os.listdir(path) if f.endswith(extension)] 

print listFiles('/Path/to/directory/with/files', '.txt') 

如果你想列出所有的在某個目錄指定的擴展及其子目錄中的文件，你可以這樣做：

import os 

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

print filterFiles('/Path/to/directory/with/files', '.txt') 

Answer 12

更改目錄到指定的路徑，這樣就可以目錄中搜索文件。如果你不改變的目錄，然後這個代碼將在您的當前目錄位置搜索文件：

import os #importing os library 
import glob #importing glob library 

path=raw_input() #input from the user 
os.chdir(path) 

filedata=glob.glob('*.c') #all files with .c extenstions stores in filedata. 
print filedata