python - 在 Python 中浏览文件和子文件夹

Question

我想浏览当前文件夹及其所有子文件夹并获取所有带有 .htm|.html 扩展名的文件。我发现可以像这样找出一个对象是目录还是文件:

import os

dirList = os.listdir("./") # current directory
for dir in dirList:
  if os.path.isdir(dir) == True:
    # I don't know how to get into this dir and do the same thing here
  else:
    # I got file and i can regexp if it is .htm|html

最后，我希望将所有文件及其路径放在一个数组中。有这样的可能吗？

Answer 1

您可以使用 os.walk()递归遍历目录及其所有子目录:

for root, dirs, files in os.walk(path):
    for name in files:
        if name.endswith((".html", ".htm")):
            # whatever

要构建这些名称的列表，您可以使用列表推导:

htmlfiles = [os.path.join(root, name)
             for root, dirs, files in os.walk(path)
             for name in files
             if name.endswith((".html", ".htm"))]