使用 Python 遍历目录

Question

我需要遍历给定目录的子目录并搜索文件。如果我得到一个文件，我必须打开它并更改内容并用我自己的行替换它。

我试过了：

import os

rootdir ='C:/Users/sid/Desktop/test'

for subdir, dirs, files in os.walk(rootdir):
    for file in files:
        f=open(file,'r')
        lines=f.readlines()
        f.close()
        f=open(file,'w')
        for line in lines:
            newline = "No you are not"
            f.write(newline)
        f.close()

但我收到错误消息。我究竟做错了什么？

Answer 1

实际浏览目录的过程与您编写的代码相同。如果你用简单的print 语句替换内部循环的内容，你可以看到每个文件都找到了：

import os
rootdir = 'C:/Users/sid/Desktop/test'

for subdir, dirs, files in os.walk(rootdir):
    for file in files:
        print os.path.join(subdir, file)

如果您在运行上述程序时仍然遇到错误，请提供错误消息。

---

为 Python3 更新

import os
rootdir = 'C:/Users/sid/Desktop/test'

for subdir, dirs, files in os.walk(rootdir):
    for file in files:
        print(os.path.join(subdir, file))

Answer 2

返回子目录中所有文件的另一种方法是使用 Python 3.4 中引入的the pathlib module，它提供了一种面向对象的方法来处理文件系统路径（Pathlib 在 Python 2.7 上也可以通过 the pathlib2 module on PyPi 获得）：

from pathlib import Path

rootdir = Path('C:/Users/sid/Desktop/test')
# Return a list of regular files only, not directories
file_list = [f for f in rootdir.glob('**/*') if f.is_file()]

# For absolute paths instead of relative the current dir
file_list = [f for f in rootdir.resolve().glob('**/*') if f.is_file()]

从 Python 3.5 开始，glob 模块也支持递归文件查找：

import os
from glob import iglob

rootdir_glob = 'C:/Users/sid/Desktop/test/**/*' # Note the added asterisks
# This will return absolute paths
file_list = [f for f in iglob(rootdir_glob, recursive=True) if os.path.isfile(f)]

上述任何一种方法中的file_list 都可以在不需要嵌套循环的情况下进行迭代：

for f in file_list:
    print(f) # Replace with desired operations

Answer 3

从 python >= 3.5 开始，您可以使用**、glob.iglob(path/**, recursive=True) 这似乎是最pythonic 的解决方案，即：

import glob, os

for filename in glob.iglob('/pardadox-music/**', recursive=True):
    if os.path.isfile(filename): # filter dirs
        print(filename)

输出：

/pardadox-music/modules/her1.mod
/pardadox-music/modules/her2.mod
...

---

注意事项：
1 - glob.iglob

glob.iglob(pathname, recursive=False)

返回一个迭代器，它产生与glob() 相同的值，但实际上并没有同时存储它们。

2 - 如果递归是True，模式'**' 将匹配任何文件和 零个或多个 directories 和 subdirectories。

3 - 如果目录包含以 . 开头的文件，则默认情况下不会匹配。例如，考虑包含card.gif 和.card.gif 的目录：

>>> import glob
>>> glob.glob('*.gif') ['card.gif'] 
>>> glob.glob('.c*')['.card.gif']

4 - 你也可以使用rglob(pattern), 这与在给定的相对模式前面添加**/ 调用glob() 相同。