Python读取指定文件夹中的所有文件的名称生成标签列表

用于机器学习：

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)

输入参数为文件夹的绝对路径，返回值为文件夹中所有文件的字符名称列表。