Python-遍历某些扩展名的文件

Question

我正在尝试遍历文件夹和所有子文件夹以查找某些文件类型的所有文件-例如，仅.mp4，.avi，.wmv。

这是我现在拥有的，它遍历所有文件类型：

import os
rootdir = 'input'

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

谢谢！

Answer 1

您可以使用os.path.splitextwhich采取路径并将文件扩展名从其末尾分割：

import os
rootdir = 'input'
extensions = ('.mp4', '.avi', '.wmv')

for subdir, dirs, files in os.walk(rootdir):
    for file in files:
        ext = os.path.splitext(file)[-1].lower()
        if ext in extensions:
            print (os.path.join(subdir, file))