Python 遍历某文件夹下所有文件夹或文件

1、使用os库中的listdir方法

os.listdir(path) 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表(不包含子文件夹里的文件夹和文件)

os.path.isfile(path)判断路径是否为文件。os.path.isdir(path)判断路径是否为目录

os.path.join(path1[, path2[, ...]])把目录和文件名合成一个路径

import os
basepath=r'根文件夹'
rootpaths=os.listdir(basepath)
for rootpath in rootpaths[:]:
    if os.path.isfile(os.path.join(basepath,rootpath)):#移除文件
        rootpaths.remove(rootpath)

2、使用os库中的scandir方法

os.scandir(path)返回指定文件夹下文件夹和文件对应的os.DirEntry对象的迭代器(不包含子文件夹里的文件夹和文件)，运行效率比os.walk高。

os.DirEntry.is_dir()，os.DirEntry对象的is_dir()方法用于检查路径是否为文件夹。

os.DirEntry.is_file()，os.DirEntry对象的is_file()方法用于检查路径是否为文件。

import os
basepath=r'根文件夹'
rootpaths=os.scandir(basepath)
rootpaths_list=[]
files_list=[]
for rootpath in rootpaths:
    if item.is_dir():#判断该对象对应路径是否为文件夹
        rootpaths_list.append(item.path)
    elif item.is_file():#判断该对象对应路径是否为文件
        files_list.append(item.path)