要求
1、遍历文件夹ddd下面所有的文件
2、并且获取所有文件所在文件夹的名称
解决方案
思路:先获取某个路径下面的其他路径,然后循环遍历,判断这个路径到底是文件路径还是文件夹路径,如果是文件路径,直接添加到集合里面,如果是文件夹的话,就递归循环遍历;
import os
##第一步、加载文件,获取文件路径以及标签
train_path="./data/ddd"
allpath=[]
lllables=[]
def get_lableandwav(path,dir):
dirs = os.listdir(path)
for a in dirs:
print(a)
print(os.path.isfile(path+"/"+a))
if os.path.isfile(path+"/"+a):
allpath.append(dirs)
if dir!="":
lllables.append(dir)
else:
get_lableandwav(str(path)+"/"+str(a),a)
##循环遍历这个文件夹
return allpath,lllables
##第一步、加载文件,获取文件路径以及标签
[allpath,lllables]=get_lableandwav(train_path,"")
print(allpath)
print("----------")
print(lllables)