使用os.walk方法获取包含两种文件类型的目录路径

debugcn 发表于 Dev

斯捷潘

我想在Python 2.7中docx使用os().walk方法列出所有包含文件的文件夹。我设法用下面的代码做到了这一点，但是我想知道是否可以限制此列表以仅显示仅包含两种特定文件类型（例如“ docx”和“ pdf”）的文件夹吗？

import os
import walk

a = open("output.txt", "w")
for path, subdirs, files in os.walk(r'C:\Users\Stephen\Desktop'):
    for filename in files:
        if filename.endswith(('.docx')):
            f = os.path.join(path, filename)
            a.write(str(f) + os.linesep)

马丁·彼得斯（Martijn Pieters）

只需跳过至少没有这两个扩展名的目录即可；每个目录的文件列表是有限的，因此any()用于测试特定扩展名的成本很低：

for path, subdirs, files in os.walk(r'C:\Users\Stephen\Desktop'):
    if not (any(f.endswith('.pdf') for f in files) and 
            any(f.endswith('.docx') for f in files)):
        # no PDF or Word files here, skip
        continue
    # directory contains *both* PDF and Word documets

当要测试的扩展列表变长时，您可能只想创建一组所有可用扩展：

for path, subdirs, files in os.walk(r'C:\Users\Stephen\Desktop'):
    extensions = {os.path.splitext(f)[-1] for f in files}
    if not extensions >= {'.pdf', '.docx', '.odt', '.wpf'}:
        # directory doesn't contain *all* required file types 
        continue

>=如果测试右侧集是左侧的一个子集（因此extensions是一个右侧集的超集）; 因此至少extensions应包含右侧命名的所有扩展名：

>>> {'.foo', '.docx', '.pdf', '.odt'} >= {'.pdf', '.docx', '.odt', '.wpf'}  # missing .wpf
False
>>> {'.foo', '.wpf', '.docx', '.pdf', '.odt'} >= {'.pdf', '.docx', '.odt', '.wpf'} # complete
True

本文收集自互联网，转载请注明来源。

如有侵权，请联系[email protected] 删除。