如何遍历给定目录中的文件?


554

我需要遍历.asm给定目录内的所有文件并对它们执行一些操作。

如何有效地做到这一点?

Answers:


807

原始答案:

import os

for filename in os.listdir(directory):
    if filename.endswith(".asm") or filename.endswith(".py"): 
         # print(os.path.join(directory, filename))
        continue
    else:
        continue

上面答案的Python 3.6版本,使用os-假设您将目录路径作为str对象包含在名为的变量中directory_in_str

import os

directory = os.fsencode(directory_in_str)

for file in os.listdir(directory):
     filename = os.fsdecode(file)
     if filename.endswith(".asm") or filename.endswith(".py"): 
         # print(os.path.join(directory, filename))
         continue
     else:
         continue

或递归使用pathlib

from pathlib import Path

pathlist = Path(directory_in_str).glob('**/*.asm')
for path in pathlist:
     # because path is object not string
     path_in_str = str(path)
     # print(path_in_str)

1
这似乎只是在目录下列出目录或文件。以下pedromateo的答案似乎是递归列出的。
杰伊·谢思

8
请注意,在Python 3.6中,目录应该以字节为单位,然后listdir也会以字节数据类型吐出文件名列表,因此您不能直接在其上运行endswith。该代码块应更改为directory = os.fsencode(directory_in_str) for file in os.listdir(directory): filename = os.fsdecode(file) if filename.endswith(".asm") or filename.endswith(".py"): # print(os.path.join(directory, filename)) continue else: continue
Kim Stacks'May

13
print(os.path.join(directory, filename))需要更改以print(os.path.join(directory_in_str, filename))使其在python 3.6中工作
Hugo Koopmans

54
如果您在2017年或以后看到它,现在可以使用os.scandir(dir_str)了,它使用起来更加简洁。无需fsencode。for entry in os.scandir(path): print(entry.path)
山羊

2
宁愿if filename.endswith((".asm", ".py")):if filename.endswith(".asm") or filename.endswith(".py"):
Maroloccio

150

这将遍历所有后代文件,而不仅仅是目录的直接子级:

import os

for subdir, dirs, files in os.walk(rootdir):
    for file in files:
        #print os.path.join(subdir, file)
        filepath = subdir + os.sep + file

        if filepath.endswith(".asm"):
            print (filepath)

3
可以在以下位置找到os.walk函数的参考:docs.python.org/2/library/os.path.html#os.path.walk
ScottMcC

136

您可以尝试使用glob模块:

import glob

for filepath in glob.iglob('my_dir/*.asm'):
    print(filepath)

从Python 3.5开始,您还可以搜索子目录:

glob.glob('**/*.txt', recursive=True) # => ['2.txt', 'sub/3.txt']

从文档:

glob模块根据Unix shell使用的规则查找与指定模式匹配的所有路径名,尽管结果以任意顺序返回。没有波浪符号扩展,但是* 、?和用[]表示的字符范围将正确匹配。


18

从Python 3.5开始,使用os.scandir()可以轻松得多

with os.scandir(path) as it:
    for entry in it:
        if entry.name.endswith(".asm") and entry.is_file():
            print(entry.name, entry.path)

使用scandir()而不是listdir()可以显着提高还需要文件类型或文件属性信息的代码的性能,因为如果操作系统在扫描目录时提供了os.DirEntry对象,则os.DirEntry对象将公开此信息。所有的os.DirEntry方法都可以执行系统调用,但是is_dir()和is_file()通常只需要系统调用即可进行符号链接。os.DirEntry.stat()在Unix上始终需要系统调用,而在Windows上只需要一个系统调用即可。


entry是一个 posix.DirEntry与一群的方便的方法,例如类型entry.is_dir()is_file()is_symlink()
crypdick

17

Python 3.4和更高版本在标准库中提供pathlib。您可以这样做:

from pathlib import Path

asm_pths = [pth for pth in Path.cwd().iterdir()
            if pth.suffix == '.asm']

或者,如果您不喜欢列表推导:

asm_paths = []
for pth in Path.cwd().iterdir():
    if pth.suffix == '.asm':
        asm_pths.append(pth)

Path 对象可以轻松转换为字符串。


9

这是我遍历Python中文件的方式:

import os

path = 'the/name/of/your/path'

folder = os.fsencode(path)

filenames = []

for file in os.listdir(folder):
    filename = os.fsdecode(file)
    if filename.endswith( ('.jpeg', '.png', '.gif') ): # whatever file types you're using...
        filenames.append(filename)

filenames.sort() # now you have the filenames and can do something with them

这些技术均无法保证任何迭代顺序

是的,超级变幻莫测。请注意,我对文件名进行了排序,这在文件顺序很重要的情况下很重要,例如,对于视频帧或与时间有关的数据收集。不过,请务必在文件名中添加索引!


并非总是排序... im1,im10,im11 ...,im2 ...否则很有用。from pkg_resources import parse_versionfilenames.sort(key=parse_version)做到了。
Hastur

5

您可以使用glob来引用目录和列表:

import glob
import os

#to get the current working directory name
cwd = os.getcwd()
#Load the images from images folder.
for f in glob.glob('images\*.jpg'):   
    dir_name = get_dir_name(f)
    image_file_name = dir_name + '.jpg'
    #To print the file name with path (path will be in string)
    print (image_file_name)

要获取数组中所有目录的列表,可以使用os

os.listdir(directory)

4

我对该实现还不太满意,我想拥有一个自定义构造函数,DirectoryIndex._make(next(os.walk(input_path)))该构造函数可以使您只传递要为其列出文件的路径。欢迎编辑!

import collections
import os

DirectoryIndex = collections.namedtuple('DirectoryIndex', ['root', 'dirs', 'files'])

for file_name in DirectoryIndex(*next(os.walk('.'))).files:
    file_path = os.path.join(path, file_name)

2

我真的很喜欢使用库中scandir内置的指令os。这是一个工作示例:

import os

i = 0
with os.scandir('/usr/local/bin') as root_dir:
    for path in root_dir:
        if path.is_file():
            i += 1
            print(f"Full path is: {path} and just the name is: {path.name}")
print(f"{i} files scanned successfully.")

重复的答案
crypdick
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.