十個Python自動化常用操作,即拿即用

一、遍歷文件夾

代碼如下,大傢可以根據自己的路徑進行修改

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)

當手上的需求明確有獲取給定路徑各級文件夾下所有符合條件的文件,並進行相應的批處理操作時,即可使用 os.walk

二、目標路徑是否是文件

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

給定一個目標路徑 path ,通過一行代碼就能夠判斷這是文件還是文件夾路徑

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

三、獲取路徑中的文件名

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

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

四、創建文件夾

創建文件夾的代碼非常常用,因為往往生成的新文件都希望有個新的文件夾存儲,代碼如下:

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

但是,如果希望創建的文件夾已經存在,再運行 os.mkdir() 則會報錯而終止代碼。為瞭避免這一情況的發生,可以在創建文件夾之前先判斷文件夾是否存在。

用到的代碼是 os.path.exists,隻有當路徑不存在(即  os.path.exists 返回的結果是 False 時),才會創建:

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

五、獲取桌面路徑

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

這樣做的好處是可以把數據放在桌面上,在不同的電腦上都能調用代碼對數據進行處理。如果是在一條電腦上把桌面路徑固定在字符串中,則換一臺電腦就必須修改桌面路徑。代碼如下:

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

當然把上面的代碼包裝成一個函數 GetDesktopPath() 需要時調用它會更加方便

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 模塊下方法可完成獲取指定路徑的全部或符合條件的文件(非遍歷各級文件夾的需求),還可以使用下面兩個代碼第一種用到的方法是os.scandir(),使用如下:

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

八、批處理文件 – 2

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

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

九、移動文件/文件夾

shutil也是經常出現在辦公自動化場景中的模塊,我常用的就是移動文件/文件夾

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

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

註意到上面後兩行代碼的區別嗎?前一行是將目標文件移動到目標文件夾裡,而後一行,在將目標文件移動到目標文件夾裡的同時,能夠對其進行重命名

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

十、批處理文件 – 3

最後要介紹的是glob模塊,也是辦公自動化必須要掌握的一個模塊,同樣可以用於批處理文件。

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

有時候我們需要對大量文件進行相同操作,在寫完針對一份文件的操作後,隻需要加上幾行代碼,就可以完成批處理全部文件的工作。大致代碼框架如下:

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

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

到此這篇關於十個Python自動化常用操作,即拿即用的文章就介紹到這瞭,更多相關Python自動化操作內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: