python關(guān)于os.walk函數(shù)查找windows文件方式

更新時(shí)間：2023年08月22日 14:43:42 作者：愿此后再無(wú)WA

這篇文章主要介紹了python關(guān)于os.walk函數(shù)查找windows文件方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

前言

OS模塊是與操作系統(tǒng)交互的模塊。這其中一個(gè)非常實(shí)際且常用的功能莫過(guò)于幫助我們查找需要的文件。

由于前面我發(fā)布過(guò)一篇比較詳細(xì)的OS模塊講解文章，這里我就不再多述。

http://m.fzitv.net/python/2959318dw.htm

直接上代碼??！

查找文件

import os
# 文件查找
def file_search():
    # os.walk的三個(gè)返回值分別是 路徑 目錄列表 文件列表  它會(huì)不斷的進(jìn)入每個(gè)目錄獲取信息
    for dir_path, listdir, files in os.walk(path):
        for file in files:
            info = dir_path + os.sep + file
            print(info)
            # 如果名字符合則用列表儲(chǔ)存起來(lái)
            # 可以把 in 換成 == 嗎？ 不建議，因?yàn)榭赡軅魅氲奈募粠Ц袷降?，但是?shí)際文件是有格式后綴的，如XX.txt
            if target_name in file:
                result.append(info)
    print()
    output()
# 文件輸出
def output():
    if result:
        print("文件出現(xiàn)在如下位置：")
        for i in result:
            print(i)
    else:
        print("未查詢(xún)到文件！")
if __name__ == '__main__':
    path = input("請(qǐng)輸入查找路徑(默認(rèn)在C盤(pán)下尋找,可直接回車(chē)):")
    path = 'C:\\' if not path else path  # 若有輸出路徑則使用給出的，否則使用默認(rèn)的C盤(pán)
    # 如果給出的路徑存在才進(jìn)行查找
    if os.path.exists(path):
        target_name = input("請(qǐng)輸入需要查找的文件名：")
        result = []
        file_search()
    else:
        print("您輸入的路徑不存在！")

其他需求

查找目錄

我們可以使用os.path模塊下的basename方法，通過(guò)路徑來(lái)截取目錄名，再看是否匹配。

那樣的話更方便，還不用看文件列表了。

def file_search():
    # os.walk的三個(gè)返回值分別是 路徑 目錄列表 文件列表  它會(huì)不斷的進(jìn)入每個(gè)目錄獲取信息
    for dir_path, listdir, files in os.walk(path):
        if os.path.basename(dir_path) == target_name:
            result.append(dir_path)
    print()
    output()

提升速度

print方法會(huì)消耗不少時(shí)間，適當(dāng)?shù)臅r(shí)候我們可以不查看當(dāng)前的路徑信息，直接等結(jié)果就好了。

另外，如果你知道有多少個(gè)文件的話可以設(shè)置一個(gè)終止條件，當(dāng)找到的文件個(gè)數(shù)滿足條件時(shí)，直接終止程序。

根據(jù)文件格式查找文件

如果知道文件格式，可以根據(jù)目標(biāo)文件的后綴匹配。

如：我想在某個(gè)找所有MP3文件或其他，就可以這么做

# 文件查找
def file_search():
    # os.walk的三個(gè)返回值分別是 路徑 目錄列表 文件列表  它會(huì)不斷的進(jìn)入每個(gè)目錄獲取信息
    for dir_path, listdir, files in os.walk(path):
        for file in files:
            info = dir_path + os.sep + file
            # endswith 以···結(jié)尾  如果要查MP3文件，那么 target_name = MP3
            if file.endswith(target_name):
                result.append(info)
    print()
    output()

模糊查找

如果你知道某個(gè)比較重要的關(guān)鍵詞可以用str1 in str2語(yǔ)句判斷是否是符合條件的文件；如果你知道更多特征可以使用正則表達(dá)式查找。

比如我想找這幾個(gè)文件，但我只知道Redis 和（）兩個(gè)關(guān)鍵字符串，那么應(yīng)該這么寫(xiě)。

# 文件查找
def file_search():
    # os.walk的三個(gè)返回值分別是 路徑 目錄列表 文件列表  它會(huì)不斷的進(jìn)入每個(gè)目錄獲取信息
    for dir_path, listdir, files in os.walk(path):
        for file in files:
            info = dir_path + os.sep + file
            #  target_name = '.*[Rr]edis.*[（(].[）)]'
            ret = re.match(target_name, file)
            if ret:
                result.append(info)
    print()
    output()

其中：

.表示任意單個(gè)字符，

*表示*前面的那個(gè)字符能夠出現(xiàn)任意多次。

.*表示任意字符出現(xiàn)任意多次。

我不知道redis的r是大寫(xiě)還是小寫(xiě)所以用[Rr]表示，[Rr]表示單個(gè)字符可選R也可選r。

不知道括號(hào)時(shí)中文還是英文的用[（(].[）)]表示。