python获取文件名的方法
不懂python获取文件名的方法?其实想解决这个问题也不难,下面让小编带着大家一起学习怎么去解决,希望大家阅读完这篇文章后大所收获。
在做项目中我们会有这个需求的,比如你训练数据过程也许会保存很多的模型,这个时候你要读取这些模型,去对测试数据进行预测,那
么你就要获取目录下的这些模型的文件名,从而读取这些模型。
方法有很多,但我还是只说一种——os.walk的方法
在这里,我再加一个特殊的需求,比如你目录下有三种模型,那么你要把这三种模型的所有文件名加入到不同的列表中,怎么办?——简
而言之,就是取出目录下的特定文件
我的model目录下有三种不同的模型,我要取出它们,分别放到不同的列表中,注意看我的注释
importos#保存不同模型的目录名(绝对路径)file_dir=r'C:\Users\zhangyh5\Desktop\xietong1012\model'bayes_list=[]#贝叶斯模型列表svm_list=[]#svm模型列表xgboost_list=[]#xgboost模型列表#root是指当前目录路径(文件夹的绝对路径)#dirs是指路径下所有的子目录(文件夹里的文件夹)#files是指路径下所有的文件(文件夹里所有的文件)forroot,dirs,filesinos.walk(file_dir):forfileinfiles:ifos.path.splitext(file)[0]=='bayes':bayes_list.append(os.path.join(root,file))elifos.path.splitext(file)[0]=='svm':svm_list.append(os.path.join(root,file))elifos.path.splitext(file)[0]=='xgboost':xgboost_list.append(os.path.join(root,file))
感谢你能够认真阅读完这篇文章,希望小编分享python获取文件名的方法内容对大家有帮助,同时也希望大家多多支持亿速云,关注亿速云行业资讯频道,遇到问题就找亿速云,详细的解决方法等着你来学习!
声明:本站所有文章资源内容,如无特殊说明或标注,均为采集网络资源。如若本站内容侵犯了原著者的合法权益,可联系本站删除。