深度学习中,常常要获取文件夹下文件名和类别,为提取训练数据和测试数据准备
本文采用的一个文件下的40类数据做实验:
最后保存成如下文件形式: 文件路径+文件名+类别标签
具体代码如下:
# -*- coding: utf-8 -*-
"""
Created on Sat Mar 21 10:51:04 2020
filenameToCSV
author: administration
"""
import os
import re
allpath=[]
allname=[]
#获取文件列表
这两天在整理一些文章,但是文件夹中每个文章没有序号会看起来很乱,所以想着能不能用Python写一个小脚本。
于是乎,参考了多方资料,简单写了下面几行代码
import osdef tekan():
i=1 #为序号赋初值
for old_file in os.listdir('.'): #os.listfir('.')用于获取当前文件夹所有文件名,'.'表示当前文件夹,也可改为目标文件路径
if 'py' not in old_file: #由于脚本文件不需要修改文件名,所以