python獲取文件名的方法：

在做項目中我們會有這個需求的，比如你訓(xùn)練數(shù)據(jù)過程也許會保存很多的模型，這個時候你要讀取這些模型，去對測試數(shù)據(jù)進行預(yù)測，那

么你就要獲取目錄下的這些模型的文件名，從而讀取這些模型。

方法有很多，但我還是只說一種——os.walk的方法

在這里，我再加一個特殊的需求，比如你目錄下有三種模型，那么你要把這三種模型的所有文件名加入到不同的列表中，怎么辦?——簡

而言之，就是取出目錄下的特定文件

我的model目錄下有三種不同的模型，我要取出它們，分別放到不同的列表中，注意看我的注釋

importos

#保存不同模型的目錄名(絕對路徑)

file_dir=r'C:\Users\zhangyh4\Desktop\xietong1012\model'

bayes_list=[]#貝葉斯模型列表

svm_list=[]#svm模型列表

xgboost_list=[]#xgboost模型列表

#root是指當(dāng)前目錄路徑(文件夾的絕對路徑)

#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))

以上內(nèi)容為大家介紹了python培訓(xùn)之如何獲取文件名，希望對大家有所幫助，如果想要了解更多Python相關(guān)知識，請關(guān)注IT培訓(xùn)機構(gòu):千鋒教育。