python獲取文件名的方法:
在做項目中我們會有這個需求的,比如你訓練數據過程也許會保存很多的模型,這個時候你要讀取這些模型,去對測試數據進行預測,那
么你就要獲取目錄下的這些模型的文件名,從而讀取這些模型。
方法有很多,但我還是只說一種——os.walk的方法
在這里,我再加一個特殊的需求,比如你目錄下有三種模型,那么你要把這三種模型的所有文件名加入到不同的列表中,怎么辦?——簡
而言之,就是取出目錄下的特定文件
我的model目錄下有三種不同的模型,我要取出它們,分別放到不同的列表中,注意看我的注釋
importos
#保存不同模型的目錄名(絕對路徑)
file_dir=r'C:\Users\zhangyh4\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培訓之如何獲取文件名,希望對大家有所幫助,如果想要了解更多Python相關知識,請關注IT培訓機構:千鋒教育。