Python: Как найти все файлы с определенным расширением?

Question

Я пытаюсь найти все файлы .c в каталоге, используя Python.

Я написал это, но он просто возвращает мне все файлы - не только .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

Try "Глоб":

>>> 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, по крайней мере с одним символом перед ним.

Edit: В качестве альтернативы, матч f[-2:] с '.c', это может быть вычислительно дешевле, чем вытащив матч регулярного выражения.

Answer 6

ПОЦЕЛУЙ

# 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 находится в документации. Я сделал это, чтобы принять список расширений для 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 (источник, Тарг, '.config', '.bat')

Answer 9

Просто чтобы быть ясно, если вы хотите, символ точки в вашем плане поиска, вы могли бы спасся тоже:

даст вам то, что вам нужно, плюс вам нужно будет использовать что-то вроде «* [обратный слэш] .c.»:

результатов.append (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