來源：早起Python

作者：劉早起

大家好，我是早起。

在過去的半年內(nèi)，我總共發(fā)布了超過了40篇 「Python辦公自動化」文章，不知道大家有沒有發(fā)現(xiàn)，有一些模塊、一些代碼會頻繁出現(xiàn)。而os、shutil、glob基本是每篇都會出現(xiàn)。

在寫了幾十篇文章后，我將一些辦公自動化常用操作的代碼單獨(dú)整理出來，大家拿走就可以用在自己的腳本中！

一、遍歷文件夾

批量操作的前提就是對文件夾進(jìn)行遍歷，使用os模塊可以輕松的遍歷文件夾，os.walk 遍歷后產(chǎn)生三個(gè)參數(shù)：

“當(dāng)前文件夾路徑包含文件夾名稱[列表形式]包含文件名稱[列表形式]”

代碼如下，大家可以根據(jù)自己的路徑進(jìn)行修改

import os
     for dirpath, dirnames, filenames in os.walk(r'C:Program Files (x86)'):
    print(f'打開文件夾{dirpath}') # 當(dāng)前文件夾路徑      if dirnames:
        print(dirnames) # 包含文件夾名稱[列表形式]      if filenames:
        print(filenames) # 包含文件名稱[列表形式]     print('-' * 10)

當(dāng)手上的需求明確有獲取給定路徑各級文件夾下所有符合條件的文件，并進(jìn)行相應(yīng)的批處理操作時(shí)，即可使用 os.walk

二、 目標(biāo)路徑是否是文件

有時(shí)我們需要判斷一個(gè)目錄下是否存在文件也可以使用os模塊。

給定一個(gè)目標(biāo)路徑 path ，通過一行代碼就能夠判斷這是文件還是文件夾路徑

import os path = 'xxx' print(os.path.isfile(path))

三、獲取路徑中的文件名

os.path.basename 可以直接從絕對路徑中獲取最后的文件名，當(dāng)然如果用傳統(tǒng)的字符串切割方式也可以，即 path.split('')[-1]

import os path = 'xxx' print(os.path.basename)

四、創(chuàng)建文件夾

創(chuàng)建文件夾的代碼非常常用，因?yàn)橥傻男挛募枷Ｍ袀€(gè)新的文件夾存儲，代碼如下：

import os dirpath = 'xxx' os.mkdir(dirpath)

但是，如果希望創(chuàng)建的文件夾已經(jīng)存在，再運(yùn)行 os.mkdir() 則會報(bào)錯(cuò)而終止代碼。為了避免這一情況的發(fā)生，可以在創(chuàng)建文件夾之前先判斷文件夾是否存在。

用到的代碼是 os.path.exists，只有當(dāng)路徑不存在（即 os.path.exists 返回的結(jié)果是 False 時(shí)），才會創(chuàng)建：

import os dirpath = 'xxx' if not os.path.exists(dirpath):
    os.mkdir(dirpath)

五、獲取桌面路徑

獲取桌面路徑也是非常常用的操作，可以使用os.path.join(os.path.expanduser("~"), 'Desktop') 獲取桌面的絕對路徑。

這樣做的好處是可以把數(shù)據(jù)放在桌面上，在不同的電腦上都能調(diào)用代碼對數(shù)據(jù)進(jìn)行處理。如果是在一條電腦上把桌面路徑固定在字符串中，則換一臺電腦就必須修改桌面路徑。代碼如下：

import os desktop_path = os.path.join(os.path.expanduser("~"), 'Desktop') print(desktop_path)

當(dāng)然把上面的代碼包裝成一個(gè)函數(shù) GetDesktopPath() 需要時(shí)調(diào)用它會更加方便

import os def GetDesktopPath():
    return os.path.join(os.path.expanduser("~"), 'Desktop')

六、重命名文件/文件夾

需要用到 os.rename() 方法，下面的代碼示例中分別演示如何重命名文件和文件夾

import os os.rename('practice.txt', 'practice_rename.txt')  # 

重命名文件 os.rename('文件夾1', '文件夾2') # 重命名文件夾

七、批處理文件 - 1

除了前面的 os.walk 之外，有其他的 os 模塊下方法可完成獲取指定路徑的全部或符合條件的文件（非遍歷各級文件夾的需求），還可以使用下面兩個(gè)代碼 第一種用到的方法是os.scandir()，使用如下：

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

八、批處理文件 - 2

上面代碼最后輸出的是 給定路徑下各內(nèi)容的名字、絕對路徑第二種方法使用 os.listdir()，它比 os.scandir() 簡單一些，可直接調(diào)用輸出名稱而非路徑：

import os  path = 'xxx' for file in os.listdir(path):
    print(file)

九、移動文件/文件夾

shutil也是經(jīng)常出現(xiàn)在辦公自動化場景中的模塊，我常用的就是移動文件/文件夾。

需要用到shutil.move 方法，下面的代碼示例中分別演示如何移動文件和文件夾：

import shutil 

shutil.move(r'.practice.txt', r'.文件夾1/') 
shutil.move(r'.practice.txt', r'.文件夾1/new.txt')

注意到上面后兩行代碼的區(qū)別嗎？前一行是將目標(biāo)文件移動到目標(biāo)文件夾里，而后一行，在將目標(biāo)文件移動到目標(biāo)文件夾里的同時(shí)，能夠?qū)ζ溥M(jìn)行重命名

也就是說，如果我們需要移動某個(gè)或某些文件到新的文件夾，并且需重命名文件，則我們并不需要用 os.rename 先命名文件再用 shutil.move 將其移動的指定文件夾，而是可以用 shutil.move 一步到位。

十、批處理文件 - 3

最后要介紹的是glob模塊，也是辦公自動化必須要掌握的一個(gè)模塊，同樣可以用于批處理文件。

glob 最重要的功能就是搜索獲取同一級或者各子級下符合條件的文件（絕對路徑），非常適合寫批處理的代碼。

有時(shí)候我們需要對大量文件進(jìn)行相同操作，在寫完針對一份文件的操作后，只需要加上幾行代碼，就可以完成批處理全部文件的工作。大致代碼框架如下：

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

glob.glob() 是一個(gè)非常重要的方法，能夠獲取給定路徑下文件的絕對路徑，并且接受「通配符」搜索，大大拓寬了靈活程度，* 表示任意字符長度，**/* 的使用表示用通配符指代給定路徑下的任何一層，recursive 參數(shù)允許遍歷搜索。

以上就是我為大家整理的十個(gè)常用操作的代碼，大部分都是可以通用的，大家可以收藏本文，并在需要進(jìn)行相關(guān)操作的時(shí)候可以直接復(fù)制粘貼使用。

另外也希望大家在自己寫代碼、讀別人代碼時(shí)，可以多去思考哪些函數(shù)/代碼片段是可以為我所用，這樣才會進(jìn)步的更快！