用python列出特定文件的问题

时间:2018-12-26 04:42:58

标签: python list file os.walk

我可以列出某些文件,但是我的代码有问题 它只能列出C:\

中的文件

我想更改代码,以便扫描其他驱动器(例如:驱动器D,E,F和...)

我该怎么做

请先检查我的代码,然后回答我的问题,以了解我的意思

我的代码:

import os

def discoverFiles(start):
    extensions = [
        'pdf','mp3','mp4','txt','docx'
    ]

    for dirpath, dirs, files in os.walk(start):
        for i in files:
            absolute_path = os.path.abspath(os.path.join(dirpath, i))
            ext = absolute_path.split('.')[-1]
            if ext in extensions:
                yield absolute_path

x = discoverFiles('/')
for j in x:
    print (j)

1 个答案:

答案 0 :(得分:0)

您可以像这样在其他驱动器中列出文件

x = discoverFiles(r'D:\\')