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