用于机器学习:
import os
def file_name(file_dir):
listName = []
#for root, dirs, files in os.walk(file_dir):
for dir in os.listdir(file_dir):
# print(dir) # 当前目录路径
listName.append(dir);
#print(dirs) # 当前路径下所有子目录
# print(files) # 当前路径下所有非目录子文件
return listName
file_dir = './imgs/test' # 待取名称文件夹的绝对路径
listname = file_name(file_dir) # listname 就是需要的标签样本
print(listname)
输入参数为文件夹的绝对路径,返回值为文件夹中所有文件的字符名称列表。