python文件名解析—从文件名获得分类类别
python os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表。列表以字母顺序。
listdir()方法
举例说明
listdir()方法
从os模块中导入函数listdir
from os import listdir返回指定文件夹中的所有文件和文件夹的名字列表
listdir(path)举例说明
文件夹dataset如下图所示，包含10个文件，每个文件的命名格式是“类别号_文件序号.txt”，每个类别号标示一类样本，文件序号表示每个类别的数量编号。例如dataset文件夹中包含10类，每个类别有一个txt文件。
分析文件名格式
文件名由‘_’和‘.’进行分割，利用split()方法对文件名进行切片。
代码示例
# -*-coding:utf-8-*-
from os import listdir
labelList = [] # 类标签列表
datasetList = listdir("dataset")
print(datasetList)
datasetLength = len(datasetList) # 文件夹中文件数量
for i in range(datasetLength):
filename = datasetList[i] # 获取文件名字符串
file = filename.split('.')[0] # 以 . 分割提取文件名
classOrder = int(file.split('_')[0]) # 以 _ 分割提取类别号
labelList.append(classOrder)
print(labelList)
