Найти все файлы в каталоге с расширением.txt в Python
Как я могу найти все файлы в директории с расширением .txt
в python?
30 ответов
можно использовать glob
:
import glob, os
os.chdir("/mydir")
for file in glob.glob("*.txt"):
print(file)
или просто os.listdir
:
import os
for file in os.listdir("/mydir"):
if file.endswith(".txt"):
print(os.path.join("/mydir", file))
или если вы хотите пересечь каталог, используйте os.walk
:
import os
for root, dirs, files in os.walk("/mydir"):
for file in files:
if file.endswith(".txt"):
print(os.path.join(root, file))
использовать Глоб.
>>> import glob
>>> glob.glob('./*.txt')
['./outline.txt', './pip-log.txt', './test.txt', './testingvim.txt']
что-то вроде этого должно сделать работу
for root, dirs, files in os.walk(directory):
for file in files:
if file.endswith('.txt'):
print file
что-то вроде этого будет работать:
>>> import os
>>> path = '/usr/share/cups/charmaps'
>>> text_files = [f for f in os.listdir(path) if f.endswith('.txt')]
>>> text_files
['euc-cn.txt', 'euc-jp.txt', 'euc-kr.txt', 'euc-tw.txt', ... 'windows-950.txt']
import os
path = 'mypath/path'
files = os.listdir(path)
files_txt = [i for i in files if i.endswith('.txt')]
мне нравится os.walk ():
import os, os.path
for root, dirs, files in os.walk(dir):
for f in files:
fullpath = os.path.join(root, f)
if os.path.splitext(fullpath)[1] == '.txt':
print fullpath
или с генераторами:
import os, os.path
fileiter = (os.path.join(root, f)
for root, _, files in os.walk(dir)
for f in files)
txtfileiter = (f for f in fileiter if os.path.splitext(f)[1] == '.txt')
for txt in txtfileiter:
print txt
вот еще несколько версий того же самого, которые дают немного разные результаты:
Глоб.iglob()
import glob
for f in glob.iglob("/mydir/*/*.txt"): # generator, search immediate subdirectories
print f
Глоб.glob1()
print glob.glob1("/mydir", "*.tx?") # literal_directory, basename_pattern
fnmatch.filter ()
import fnmatch, os
print fnmatch.filter(os.listdir("/mydir"), "*.tx?") # include dot-files
path.py есть и другая альтернатива:https://github.com/jaraco/path.py
from path import path
p = path('/path/to/the/directory')
for f in p.files(pattern='*.txt'):
print f
Python имеет все инструменты для этого:
import os
the_dir = 'the_dir_that_want_to_search_in'
all_txt_files = filter(lambda x: x.endswith('.txt'), os.listdir(the_dir))
import os
import sys
if len(sys.argv)==2:
print('no params')
sys.exit(1)
dir = sys.argv[1]
mask= sys.argv[2]
files = os.listdir(dir);
res = filter(lambda x: x.endswith(mask), files);
print res
вы можете просто использовать pathlib
s glob
1:
import pathlib
list(pathlib.Path('your_directory').glob('*.txt'))
или в цикле:
for txt_file in pathlib.Path('your_directory').glob('*.txt'):
# do something with "txt_file"
если вы хотите, чтобы он рекурсивный, вы можете использовать .glob('**/*.txt)
1на pathlib
модуль был включен в стандартную библиотеку в Python 3.4. Но вы можете установить back-порты этого модуля даже на более старые версии Python (т. е. используя conda
или pip
):pathlib
и pathlib2
.
этот код делает мою жизнь проще.
import os
fnames = ([file for root, dirs, files in os.walk(dir)
for file in files
if file.endswith('.txt') #or file.endswith('.png') or file.endswith('.pdf')
])
for fname in fnames: print(fname)
вы можете попробовать этот код
import glob
import os
filenames_without_extension = [os.path.basename(c).split('.')[0:1][0] for c in glob.glob('your/files/dir/*.txt')]
filenames_with_extension = [os.path.basename(c) for c in glob.glob('your/files/dir/*.txt')]
использовать fnmatch:https://docs.python.org/2/library/fnmatch.html
import fnmatch
import os
for file in os.listdir('.'):
if fnmatch.fnmatch(file, '*.txt'):
print file
Я сделал тест (Python 3.6.4, W7x64), чтобы увидеть, какое решение является самым быстрым для одной папки, без подкаталогов, чтобы получить список полных путей к файлам с определенным расширением.
чтобы сделать его коротким, для этой задачи os.listdir()
является самым быстрым и 1.7 x так же быстро, как следующий лучший:os.walk()
(с перерывом!), 2.7 x так же быстро, как pathlib
, 3.2 x быстрее, чем os.scandir()
и 3.3 x быстрее, чем glob
.
Пожалуйста, имейте в виду, что эти результаты изменятся, когда вам понадобится рекурсивный результаты. Если вы копируете / вставляете один метод ниже, добавьте a .ниже() иначе .EXT не будет найден при поиске .доб.
import os
import pathlib
import timeit
import glob
def a():
path = pathlib.Path().cwd()
list_sqlite_files = [str(f) for f in path.glob("*.sqlite")]
def b():
path = os.getcwd()
list_sqlite_files = [f.path for f in os.scandir(path) if os.path.splitext(f)[1] == ".sqlite"]
def c():
path = os.getcwd()
list_sqlite_files = [os.path.join(path, f) for f in os.listdir(path) if f.endswith(".sqlite")]
def d():
path = os.getcwd()
os.chdir(path)
list_sqlite_files = [os.path.join(path, f) for f in glob.glob("*.sqlite")]
def e():
path = os.getcwd()
list_sqlite_files = [os.path.join(path, f) for f in glob.glob1(str(path), "*.sqlite")]
def f():
path = os.getcwd()
list_sqlite_files = []
for root, dirs, files in os.walk(path):
for file in files:
if file.endswith(".sqlite"):
list_sqlite_files.append( os.path.join(root, file) )
break
print(timeit.timeit(a, number=1000))
print(timeit.timeit(b, number=1000))
print(timeit.timeit(c, number=1000))
print(timeit.timeit(d, number=1000))
print(timeit.timeit(e, number=1000))
print(timeit.timeit(f, number=1000))
результаты:
# Python 3.6.4
0.431
0.515
0.161
0.548
0.537
0.274
чтобы получить все '.txt "имена файлов внутри папки "dataPath" в виде списка в Pythonic way
from os import listdir
from os.path import isfile, join
path = "/dataPath/"
onlyTxtFiles = [f for f in listdir(path) if isfile(join(path, f)) and f.endswith(".txt")]
print onlyTxtFiles
import glob,os
data_dir = 'data_folder/'
file_dir_extension = os.path.join(data_dir, '*.txt')
for file_name in glob.glob(file_dir_extension):
if file_name.endswith('.txt'):
print file_name
для меня. Это классика.
Я предлагаю вам использовать fnmatch и верхний метод. Таким образом, вы можете найти следующее:
- имя.txt;
- имя.TXT;
- имя.Txt
.
import fnmatch
import os
for file in os.listdir("/Users/Johnny/Desktop/MyTXTfolder"):
if fnmatch.fnmatch(file.upper(), '*.TXT'):
print(file)
Python v3.5+
быстрый метод с использованием ОС.scandir в рекурсивной функции. Выполняет поиск всех файлов с указанным расширением в папках и подпапках.
import os
def findFilesInFolder(path, pathList, extension, subFolders = True):
""" Recursive function to find all files of an extension type in a folder (and optionally in all subfolders too)
path: Base directory to find files
pathList: A list that stores all paths
extension: File extension to find
subFolders: Bool. If True, find files in all subfolders under path. If False, only searches files in the specified folder
"""
try: # Trapping a OSError: File permissions problem I believe
for entry in os.scandir(path):
if entry.is_file() and entry.path.endswith(extension):
pathList.append(entry.path)
elif entry.is_dir() and subFolders: # if its a directory, then repeat process as a nested function
pathList = findFilesInFolder(entry.path, pathList, extension, subFolders)
except OSError:
print('Cannot access ' + path +'. Probably a permissions error')
return pathList
dir_name = r'J:\myDirectory'
extension = ".txt"
pathList = []
pathList = findFilesInFolder(dir_name, pathList, extension, True)
функциональное решение с подкаталогами:
from fnmatch import filter
from functools import partial
from itertools import chain
from os import path, walk
print(*chain(*(map(partial(path.join, root), filter(filenames, "*.txt")) for root, _, filenames in walk("mydir"))))
если папка содержит много файлов или память является ограничением, рассмотрите возможность использования генераторов:
def yield_files_with_extensions(folder_path, file_extension):
for _, _, files in os.walk(folder_path):
for file in files:
if file.endswith(file_extension):
yield file
Вариант A: Повторите
for f in yield_files_with_extensions('.', '.txt'):
print(f)
вариант B: получить все
files = [f for f in yield_files_with_extensions('.', '.txt')]
import glob
import os
path=os.getcwd()
extensions=('*.py','*.cpp')
for i in extensions:
for files in glob.glob(i):
print files
попробуйте это это найдет весь ваш файл внутри папки или папки
import glob, os
os.chdir("H:\wallpaper")# use whatever you directory
#double\ no single \
for file in glob.glob("**/*.psd", recursive = True):#your format
print(file)
чтобы получить массив ".txt " имена файлов из папки под названием "данные" в том же каталоге я обычно использую эту простую строку кода:
import os
fileNames = [fileName for fileName in os.listdir("data") if fileName.endswith(".txt")]
копия-pastable решение похоже на один из ghostdog:
def get_all_filepaths(root_path, ext):
"""
Search all files which have a given extension within root_path.
This ignores the case of the extension and searches subdirectories, too.
Parameters
----------
root_path : str
ext : str
Returns
-------
list of str
Examples
--------
>>> get_all_filepaths('/run', '.lock')
['/run/unattended-upgrades.lock',
'/run/mlocate.daily.lock',
'/run/xtables.lock',
'/run/mysqld/mysqld.sock.lock',
'/run/postgresql/.s.PGSQL.5432.lock',
'/run/network/.ifstate.lock',
'/run/lock/asound.state.lock']
"""
import os
all_files = []
for root, dirs, files in os.walk(root_path):
for filename in files:
if filename.lower().endswith(ext):
all_files.append(os.path.join(root, filename))
return all_files
- вот extend()
types = ('*.jpg', '*.png')
images_list = []
for files in types:
images_list.extend(glob.glob(os.path.join(path, files)))
вы можете попробовать этот код:
import glob
import os
os.chdir("D:\...\DirName")
filename_arr={}
i=0
for files in glob.glob("*.txt"):
filename_arr[i] = files
i= i+1
for key,value in filename_arr.items():
print key , value
использовать Python OS модуль для поиска файлов с определенным расширением.
простой пример здесь :
import os
# This is the path where you want to search
path = r'd:'
# this is extension you want to detect
extension = '.txt' # this can be : .jpg .png .xls .log .....
for root, dirs_list, files_list in os.walk(path):
for file_name in files_list:
if os.path.splitext(file_name)[-1] == extension:
file_name_path = os.path.join(root, file_name)
print file_name
print file_name_path # This is the full path of the filter file
многие пользователи ответили с os.walk
ответы, который включает в себя все файлы, но и все каталоги и подкаталоги и их файлы.
import os
def files_in_dir(path, extension=''):
"""
Generator: yields all of the files in <path> ending with
<extension>
\param path Absolute or relative path to inspect,
\param extension [optional] Only yield files matching this,
\yield [filenames]
"""
for _, dirs, files in os.walk(path):
dirs[:] = [] # do not recurse directories.
yield from [f for f in files if f.endswith(extension)]
# Example: print all the .py files in './python'
for filename in files_in_dir('./python', '*.py'):
print("-", filename)
или для одного, где вам не нужен генератор:
path, ext = "./python", ext = ".py"
for _, _, dirfiles in os.walk(path):
matches = (f for f in dirfiles if f.endswith(ext))
break
for filename in matches:
print("-", filename)
если вы собираетесь использовать совпадения для чего-то еще, вы можете сделать это списком, а не выражением генератора:
matches = [f for f in dirfiles if f.endswith(ext)]
простой метод, с помощью for
петли :
import os
dir = ["e","x","e"]
p = os.listdir('E:') #path
for n in range(len(p)):
name = p[n]
myfile = [name[-3],name[-2],name[-1]] #for .txt
if myfile == dir :
print(name)
else:
print("nops")
хотя это можно сделать более обобщенным .