用python扫描文件目录

1 使用

通过os.walk()扫描制定目录下的所有文件以及子文件夹

'''
扫描目录下的第一层文件，不包括子文件夹中的文件
'''
def get_dir_lists(root_dir):
    root_dir = r"J:\电子书"
    dir_tree = os.walk(root_dir)
    sub_dir = dir_tree.__next__()[1]
    file_list = []
    for sub in sub_dir:
        for _sub in os.walk(root_dir + "\\" + str(sub)):
            _book_list = _sub[-1]
            if _sub[1] is not None:
                for book in _book_list:
                    if ".pdf" in book:
                        file_list.append(book)
    return file_list

写在最后

欢迎大家关注鄙人的公众号【麦田里的守望者zhg】，让我们一起成长，谢谢。