python獲取文件名的方法

不懂python獲取文件名的方法？其實(shí)想解決這個(gè)問題也不難，下面讓小編帶著大家一起學(xué)習(xí)怎么去解決，希望大家閱讀完這篇文章后大所收獲。

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

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

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

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

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

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

import os
# 保存不同模型的目錄名(絕對路徑)
file_dir = r'C:\Users\zhangyh5\Desktop\xietong1012\model'
bayes_list = []     # 貝葉斯模型列表
svm_list = []        # svm模型列表
xgboost_list = []  # xgboost模型列表
# root是指當(dāng)前目錄路徑(文件夾的絕對路徑)
# dirs是指路徑下所有的子目錄(文件夾里的文件夾)
# files是指路徑下所有的文件(文件夾里所有的文件)
for root,dirs,files in os.walk(file_dir):
    for file in files:
        if os.path.splitext(file)[0] == 'bayes':
            bayes_list.append(os.path.join(root,file))
        elif os.path.splitext(file)[0] == 'svm':
            svm_list.append(os.path.join(root,file))
        elif os.path.splitext(file)[0] == 'xgboost':
            xgboost_list.append(os.path.join(root,file))

感謝你能夠認(rèn)真閱讀完這篇文章，希望小編分享python獲取文件名的方法內(nèi)容對大家有幫助，同時(shí)也希望大家多多支持億速云，關(guān)注億速云行業(yè)資訊頻道，遇到問題就找億速云，詳細(xì)的解決方法等著你來學(xué)習(xí)!