1、使用os库中的listdir方法
os.listdir(path) 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表(不包含子文件夹里的文件夹和文件)
os.path.isfile(path)判断路径是否为文件。os.path.isdir(path)判断路径是否为目录
os.path.join(path1[, path2[, ...]])把目录和文件名合成一个路径
python
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()方法用于检查路径是否为文件。
python
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)