Як я можу знайти всі файли в каталозі, що має розширення .txt
в python?
Як я можу знайти всі файли в каталозі, що має розширення .txt
в python?
Відповіді:
Ви можете використовувати 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))
for file in f
ніж для, for files in f
оскільки те, що є в змінній, - це одне ім'я файлу. Ще краще було б змінити f
на, files
і тоді петлі для циклу можуть стати for file in files
.
file
це не зарезервоване слово, а лише назва заздалегідь визначеної функції, тому цілком можливо використовувати його як ім'я змінної у власному коді. Хоча це правда, що, як правило, слід уникати подібних зіткнень, file
це особливий випадок, оскільки навряд чи коли-небудь виникає потреба у використанні, тому часто вважають винятком із настанови. Якщо ви не хочете цього робити, PEP8 рекомендує до таких імен додати єдине підкреслення, тобто file_
, з яким вам доведеться погодитися, все ще досить читабельно.
Використовуйте глобус .
>>> import glob
>>> glob.glob('./*.txt')
['./outline.txt', './pip-log.txt', './test.txt', './testingvim.txt']
glob
щоб не вдалося знайти файли рекурсивно, якщо ваш пітон менше 3,5. докладніше
Щось подібне повинно зробити цю роботу
for root, dirs, files in os.walk(directory):
for file in files:
if file.endswith('.txt'):
print file
root, dirs, files
замість r, d, f
. Набагато читабельніший.
Щось подібне спрацює:
>>> 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']
os.path.join
для кожного елемента text_files
. Це могло бути щось на кшталт text_files = [os.path.join(path, f) for f in os.listdir(path) if f.endswith('.txt')]
.
Ви можете просто використовувати pathlib
s 1 :glob
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)
1pathlib
модуль був включений в стандартній бібліотеці в Python 3.4. Але ви можете встановити резервні порти цього модуля навіть у старих версіях Python (тобто, використовуючи conda
або pip
): pathlib
і pathlib2
.
**/*.txt
не підтримується старішими версіями python. Тому я вирішив це за допомогою: foundfiles= subprocess.check_output("ls **/*.txt", shell=True)
for foundfile in foundfiles.splitlines():
print foundfile
pathlib
можна зробити, і я вже включив вимоги до версії Python. :) Але якщо ваш підхід уже не розміщено, чому б просто не додати його як іншу відповідь?
rglob
якщо ви хочете шукати предмети рекурсивно. Напр..rglob('*.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
for root, dirs, files in os.walk(dir):
for f in files:
if os.path.splitext(f)[1] == '.txt':
fullpath = os.path.join(root, f)
print(fullpath)
Або з генераторами:
import os
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)
Ось додаткові версії, які дають дещо інші результати:
import glob
for f in glob.iglob("/mydir/*/*.txt"): # generator, search immediate subdirectories
print f
print glob.glob1("/mydir", "*.tx?") # literal_directory, basename_pattern
import fnmatch, os
print fnmatch.filter(os.listdir("/mydir"), "*.tx?") # include dot-files
glob1()
- це допоміжна функція в glob
модулі, яка не вказана в документації Python. Деякі вбудовані коментарі, що описують, що він робить у вихідному файлі, див .../Lib/glob.py
.
glob.glob1()
не є загальнодоступним, але він доступний на Python 2.4-2.7; 3.0-3.2; піпі; jython github.com/zed/test_glob1
glob
модуля.
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
for f in p.walk(pattern='*.txt')
усіх папок
list(p.glob('**/*.py'))
Швидкий метод, що використовує os.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)
Якщо ви шукаєте в каталогах, що містять 10 000s файлів, додавання до списку стає неефективним. «Поступаючись» результатам - це краще рішення. Я також включив функцію перетворення результату в Datandreme Pandas.
import os
import re
import pandas as pd
import numpy as np
def findFilesInFolderYield(path, extension, containsTxt='', subFolders = True, excludeText = ''):
""" 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
extension: File extension to find. e.g. 'txt'. Regular expression. Or 'ls\d' to match ls1, ls2, ls3 etc
containsTxt: List of Strings, only finds file if it contains this text. Ignore if '' (or blank)
subFolders: Bool. If True, find files in all subfolders under path. If False, only searches files in the specified folder
excludeText: Text string. Ignore if ''. Will exclude if text string is in path.
"""
if type(containsTxt) == str: # if a string and not in a list
containsTxt = [containsTxt]
myregexobj = re.compile('\.' + extension + '$') # Makes sure the file extension is at the end and is preceded by a .
try: # Trapping a OSError or FileNotFoundError: File permissions problem I believe
for entry in os.scandir(path):
if entry.is_file() and myregexobj.search(entry.path): #
bools = [True for txt in containsTxt if txt in entry.path and (excludeText == '' or excludeText not in entry.path)]
if len(bools)== len(containsTxt):
yield entry.stat().st_size, entry.stat().st_atime_ns, entry.stat().st_mtime_ns, entry.stat().st_ctime_ns, entry.path
elif entry.is_dir() and subFolders: # if its a directory, then repeat process as a nested function
yield from findFilesInFolderYield(entry.path, extension, containsTxt, subFolders)
except OSError as ose:
print('Cannot access ' + path +'. Probably a permissions error ', ose)
except FileNotFoundError as fnf:
print(path +' not found ', fnf)
def findFilesInFolderYieldandGetDf(path, extension, containsTxt, subFolders = True, excludeText = ''):
""" Converts returned data from findFilesInFolderYield and creates and Pandas Dataframe.
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
extension: File extension to find. e.g. 'txt'. Regular expression. Or 'ls\d' to match ls1, ls2, ls3 etc
containsTxt: List of Strings, only finds file if it contains this text. Ignore if '' (or blank)
subFolders: Bool. If True, find files in all subfolders under path. If False, only searches files in the specified folder
excludeText: Text string. Ignore if ''. Will exclude if text string is in path.
"""
fileSizes, accessTimes, modificationTimes, creationTimes , paths = zip(*findFilesInFolderYield(path, extension, containsTxt, subFolders))
df = pd.DataFrame({
'FLS_File_Size':fileSizes,
'FLS_File_Access_Date':accessTimes,
'FLS_File_Modification_Date':np.array(modificationTimes).astype('timedelta64[ns]'),
'FLS_File_Creation_Date':creationTimes,
'FLS_File_PathName':paths,
})
df['FLS_File_Modification_Date'] = pd.to_datetime(df['FLS_File_Modification_Date'],infer_datetime_format=True)
df['FLS_File_Creation_Date'] = pd.to_datetime(df['FLS_File_Creation_Date'],infer_datetime_format=True)
df['FLS_File_Access_Date'] = pd.to_datetime(df['FLS_File_Access_Date'],infer_datetime_format=True)
return df
ext = 'txt' # regular expression
containsTxt=[]
path = 'C:\myFolder'
df = findFilesInFolderYieldandGetDf(path, ext, containsTxt, subFolders = True)
У 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))
all_txt_files = list(filter(lambda x: x.endswith('.txt'), os.listdir(the_dir)))
Спробуйте це. Ви знайдете всі ваші файли рекурсивно:
import glob, os
os.chdir("H:\\wallpaper")# use whatever directory you want
#double\\ no single \
for file in glob.glob("**/*.txt", recursive = True):
print(file)
**
. Доступний лише в python 3. Те, що мені не подобається, це chdir
частина. Не потрібно в цьому.
filepath = os.path.join('wallpaper')
а потім використовувати його як glob.glob(filepath+"**/*.psd", recursive = True)
, що дасть такий же результат.
Я зробив тест (Python 3.6.4, W7x64), щоб побачити, яке рішення є найшвидшим для однієї папки, без підкаталогів, щоб отримати список повних шляхів до файлів для файлів із конкретним розширенням.
Якщо коротко, це завдання os.listdir()
найшвидше і на 1,7 рази швидше, ніж наступне найкраще: os.walk()
(з перервою!), 2,7 разів швидше pathlib
, на 3,2 os.scandir()
рази швидше і на 3,3 рази швидше glob
.
Зауважте, що ці результати змінюватимуться, коли вам знадобляться рекурсивні результати. Якщо ви скопіюєте / вставте один із способів нижче, додайте .lower (), інакше .EXT не буде знайдено під час пошуку .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
Цей код робить моє життя простішим.
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)
Використовуйте 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
Щоб отримати масив імен файлів ".txt" з папки під назвою "data" в одному каталозі, я зазвичай використовую цей простий рядок коду:
import os
fileNames = [fileName for fileName in os.listdir("data") if fileName.endswith(".txt")]
Я пропоную вам використовувати fnmatch та верхній метод. Таким чином ви можете знайти будь-що з наступного:
.
import fnmatch
import os
for file in os.listdir("/Users/Johnny/Desktop/MyTXTfolder"):
if fnmatch.fnmatch(file.upper(), '*.TXT'):
print(file)
Функціональне рішення з підкаталогами:
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
Варіант А: Ітерація
for f in yield_files_with_extensions('.', '.txt'):
print(f)
Варіант В: Отримайте все
files = [f for f in yield_files_with_extensions('.', '.txt')]
Копіювальне рішення, схоже на рішення привида:
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
використовуйте модуль 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")
Хоча це можна зробити більш узагальненим.