作者：陳熹 

來源：早起Python

大家好，又到了python辦公自動化系列。在日常的辦公中，我們經常會從一堆不同格式的文件(夾)中搜索特定的文件，可能你是憑著記憶去找或是借助軟件，但你有想過如何用python實現(xiàn)嗎？

掃描路徑內的內容

有些時候我們會希望在當前文件夾的成百上千個文件中快速找到需要的文件，如果這個文件夾又包括很多個子文件夾，并不需要程序進去查找而無端耗費資源。這就是典型的「掃描一層搜索」

1. 基于 os.scandir()

需求說明

?

輸出C:\Program Files (x86)路徑下名字包含“Windows”的文件夾名稱，并統(tǒng)計個數

?

步驟分析

這是一個非遍歷的需求，只需要在目的文件夾內掃描一圈判斷即可。用到的方法是os.scandir()，使用如下：

import os
path = ...
for file in os.scandir(path): 
    print(file.name, file.path, file.is_dir())

上面代碼最后輸出的是給定路徑下各內容的名字、絕對路徑，并判斷其是否是文件夾

需要注意的是os.scandir()只在路徑下一級掃描，需求實現(xiàn)的代碼如下(代碼邏輯很簡單，可以自己做相應調整)：

import os
num = 0
for file in os.scandir(r'C:\\Program Files (x86)'):
    if file.is_dir():
        if 'Windows' in file.name:
            print(file.name)
            num += 1
print('含有Windows的文件夾個數為：', num)

2. 基于 os.listdir()

需求說明

輸出C:\Program Files (x86)路徑下所有可執(zhí)行文件 (即后綴為 .exe)

步驟分析

依然是非遍歷的需求，這里使用os.listdir()，它比os.scandir()簡單一些，可直接調用輸出名稱而非路徑。后綴名可以用字符串的切片來判斷，但難免失去靈活性。此時建議用字符串方法string.endswith()來判斷名稱的結尾是否是.exe，代碼如下：

import os 
for file in os.listdir(r'C:\\Program Files (x86)'):
    if file.endswith('.exe'):
        print(file)

三、遍歷文件夾搜索文件

更多時候我們希望給定一個大概的路徑，在這個路徑下的所有文件夾里一層一層找，找到特定文件或者符合要求的文件，這里需要遍歷文件，有兩種主要的方法：

1. 基于 os.walk()

os.walk 遍歷后產生三個參數：當前文件夾路徑, 包含文件夾名稱  [列表形式], 包含文件名稱[列表形式]，可以用如下代碼完成簡單遍歷：

import os
for dirpath, dirnames, filenames in os.walk(r'C:\\Program Files (x86)'):
    print(f'打開文件夾{dirpath}')
    if dirnames:
        print(dirnames)
    if filenames:
        print(filenames)
    print('-' * 10)

需求說明

遍歷C:\Program Files (x86)找出所有新版Excel文件 (即后綴為.xlsx)

步驟分析

只要理解了os.walk()的工作模式，用endswith()判斷后綴即可。最后如果需要獲得絕對路徑可以把當前文件夾路徑和文件名拼接，簡單一點用 + 或者字符串格式化，也可以用os模塊內的方法

import os
for dirpath, dirnames, filenames in os.walk(r'C:\\Program Files (x86)'):
    if filenames:
        for i in filenames:
            if i.endswith('.xlsx'):
                print(os.path.join(dirpath, i))

2. 基于非 os 方法：glob

glob在之前的推文也反復提到遍歷框架：

import glob
for file in glob.glob('**/*', recursive=True): 
    print(file)

**/*的使用表示用通配符指代給定路徑下的任何一層，recursive參數允許遍歷搜索 由于glob可以使用「通配符」，大大拓寬了靈活程度，這里利用 glob 也來完成上面的需求

需求說明

遍歷C:\Program Files (x86)找出所有新版Excel文件(即后綴為.xlsx)

import glob
for file in glob.glob('**/*.xlsx', recursive=True): 
    print(file)

可以看到非常簡單，在原有代碼基礎上加上后綴名就能夠完成特定類型文件的搜索。如果需要搜索「特定文件」，如遍歷C:\Program Files (x86)找到文件practice.txt，「只需要后一個*改成具體名稱就行」

import glob
for file in glob.glob('**/practice.txt', recursive=True): 
    print(file)

四、寫在最后

通過本文的Python自動化腳本制作過程，我們可以再次體會Python辦公自動化的強大！如果對自動化代碼和數據感興趣可以在后臺回復自動化獲取。

當然本文只是基于幾個簡單的需求來講解使用Python來制作搜索文件腳本的主要方法，接下來你可以結合之前的自動化案例或以根據自己的需求添加一些其他的規(guī)則或增加新的功能，打造出一款屬于你的軟件！

最后還是希望大家能夠理解Python辦公自動化的一個核心就是「批量操作-解放雙手」，讓復雜的工作自動化！