Python에서 확장자가.txt 인 디렉토리의 모든 파일 찾기




파이썬 확장자명 (24)

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

어떻게 확장자가 .txt 디렉토리에있는 모든 파일을 파이썬에서 찾을 수 있습니까?


나는 os.walk() 좋아 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

import glob
import os

path=os.getcwd()

extensions=('*.py','*.cpp')

for i in extensions:
  for files in glob.glob(i):
     print 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

이 코드를 사용해보십시오.

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')]

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

동일한 디렉토리에서 "data"라는 폴더의 ".txt"파일 이름 배열을 얻으려면 다음과 같은 간단한 코드 행을 사용합니다.

import os
fileNames = [fileName for fileName in os.listdir("data") if fileName.endswith(".txt")]

이것을 시도하면 폴더 나 폴더 안에있는 모든 파일을 찾을 수 있습니다.

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)

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,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

나를 위해서. 그것은 고전적입니다.


Python v3.5 이상

재귀 함수에서 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)


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

많은 사용자가 모든 파일뿐만 아니라 모든 디렉토리와 하위 디렉토리 및 해당 파일을 포함하는 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)]

이 코드는 제 인생을 단순하게 만듭니다.

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 os

the_dir = 'the_dir_that_want_to_search_in'
all_txt_files = filter(lambda x: x.endswith('.txt'), os.listdir(the_dir))

폴더에 많은 파일이 있거나 메모리가 제약 조건 인 경우 발전기 사용을 고려하십시오.

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')]

하위 디렉토리가있는 기능적 솔루션 :

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"))))

glob 사용하십시오.

>>> import glob
>>> glob.glob('./*.txt')
['./outline.txt', './pip-log.txt', './test.txt', './testingvim.txt']

어떤 솔루션이 하나의 폴더 (하위 디렉토리 없음)에 대해 가장 빠른 솔루션인지 확인하기위한 테스트 (Python 3.6.4, W7x64)를 수행하여 특정 확장자를 가진 파일의 전체 파일 경로 목록을 가져 왔습니다.

이것을 짧게하기 위해, os.listdir() 은 가장 빠르며 차선책보다 1.7 배 빠릅니다 : os.walk() (break!), pathlib 2.7 배, 3.2 배의 속도 os.scandir()glob 보다 3.3 배 빠릅니다.
재귀 결과가 필요할 때 이러한 결과가 변경된다는 점에 유의하십시오. 아래의 한 가지 방법을 복사 / 붙여 넣기하려면 .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

path = 'mypath/path' 
files = os.listdir(path)

files_txt = [i for i in files if i.endswith('.txt')]

약간 다른 결과가 나오는 더 많은 버전이 있습니다.

glob.iglob()

import glob
for f in glob.iglob("/mydir/*/*.txt"): # generator, search immediate subdirectories 
    print f

glob.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

여기에 extend()

types = ('*.jpg', '*.png')
images_list = []
for files in types:
    images_list.extend(glob.glob(os.path.join(path, files)))

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")

이것은 더 일반화 될 수 있지만.





file-io