我试图在一个文件夹和所有子文件夹中循环查找特定文件类型的所有文件——例如,仅限.mp4.阿维.wmv.
以下是我现在拥有的,它循环遍历所有文件类型:
import os
rootdir = 'input'
for subdir, dirs, files in os.walk(rootdir):
for file in files:
print (os.path.join(subdir, file))
我试图在一个文件夹和所有子文件夹中循环查找特定文件类型的所有文件——例如,仅限.mp4.阿维.wmv.
以下是我现在拥有的,它循环遍历所有文件类型:
import os
rootdir = 'input'
for subdir, dirs, files in os.walk(rootdir):
for file in files:
print (os.path.join(subdir, file))
您可以使用os.path.splitext
,它获取一个路径,并将文件扩展名从其末尾拆分:
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))