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

更新時間：2023年08月22日 14:43:42 作者：愿此后再無WA

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

前言

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

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

http://www.dbjr.com.cn/python/2959318dw.htm

直接上代碼??！

查找文件

import os
# 文件查找
def file_search():
    # os.walk的三個返回值分別是 路徑 目錄列表 文件列表  它會不斷的進入每個目錄獲取信息
    for dir_path, listdir, files in os.walk(path):
        for file in files:
            info = dir_path + os.sep + file
            print(info)
            # 如果名字符合則用列表儲存起來
            # 可以把 in 換成 == 嗎？ 不建議，因為可能傳入的文件名不帶格式的，但是實際文件是有格式后綴的，如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("未查詢到文件！")
if __name__ == '__main__':
    path = input("請輸入查找路徑(默認在C盤下尋找,可直接回車):")
    path = 'C:\\' if not path else path  # 若有輸出路徑則使用給出的，否則使用默認的C盤
    # 如果給出的路徑存在才進行查找
    if os.path.exists(path):
        target_name = input("請輸入需要查找的文件名：")
        result = []
        file_search()
    else:
        print("您輸入的路徑不存在！")

其他需求

查找目錄

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

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

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

提升速度

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

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

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

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

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

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

模糊查找

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

比如我想找這幾個文件，但我只知道Redis 和（）兩個關鍵字符串，那么應該這么寫。

# 文件查找
def file_search():
    # os.walk的三個返回值分別是 路徑 目錄列表 文件列表  它會不斷的進入每個目錄獲取信息
    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()

其中：

.表示任意單個字符，

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

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

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

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