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)