快捷導(dǎo)航

Python實現(xiàn)提取指定名稱的文件并批量復(fù)制到其他文件夾

更新時間：2023年10月19日 11:04:07 作者：瘋狂學習GIS

本文介紹基于Python語言,讀取一個文件夾,并將其中每一個子文件夾內(nèi)符合名稱要求的文件加以篩選,并將篩選得到的文件復(fù)制到另一個目標文件夾中的方法,需要的朋友可以參考下

本文的需求是：現(xiàn)在有一個大的文件夾，其中含有多個子文件夾，如下圖所示。

對于其中的每一個子文件夾，都含有大量的文件；例如，我們就打開上圖中的2021文件夾，得到結(jié)果如下圖所示。

我們的需求是，希望基于每一個子文件夾內(nèi)的每一個文件，將文件名稱中含有指定字段的文件提取出來——在本文中，我們就希望提取得到文件名稱中含有STB字段的，也就是上圖紫色框內(nèi)出現(xiàn)STB的文件是我們需要的。提取到全部符合要求的文件后，我們還需要將這些文件都復(fù)制到另一個目標文件夾中。

明確了需求，接下來就可以開始代碼的撰寫。本文所用到的代碼如下所示。

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Mon Oct  2 12:48:49 2023

@author: fkxxgis
"""

import os
import shutil

source_folder = "/media/dell/H30-35/GF_SR"
target_folder = "/media/dell/H30-35/GF_SR/STB"

if not os.path.exists(target_folder):
    os.makedirs(target_folder)

year_folders = [f for f in os.listdir(source_folder) if os.path.isdir(os.path.join(source_folder, f))]

for year_folder in year_folders:
    if year_folder == "STB":
        continue
    year_folder_path = os.path.join(source_folder, year_folder)
    for file_name in os.listdir(year_folder_path):
        if "STB" in file_name:
            source_file = os.path.join(year_folder_path, file_name)
            target_file = os.path.join(target_folder, file_name)
            shutil.copy(source_file, target_file)

其中，我們導(dǎo)入了os和shutil模塊，它們都提供了處理文件和目錄的功能；同時，我們定義了源文件夾路徑source_folder（也就是存儲有多個子文件夾的文件夾路徑）和目標文件夾路徑target_folder（也就是最終結(jié)果存放的路徑）。

隨后，我們使用os.makedirs()函數(shù)檢查目標文件夾是否存在，如果不存在則創(chuàng)建它。接下來，我們使用列表推導(dǎo)式生成一個包含源文件夾中所有子文件夾的列表year_folders；其中，我們使用了os.listdir()函數(shù)獲取源文件夾中的所有文件和文件夾，并使用os.path.isdir()函數(shù)判斷是否為文件夾。

接下來，我們就開始遍歷每個子文件夾。首先，需要注意的是，我們要檢查當前遍歷的子文件夾是否為我們的目標文件夾路徑STB；如果是，則跳過該子文件夾的處理——這一步驟是防止我們用來存放結(jié)果的目標文件夾也被遍歷，否則就會出現(xiàn)文件復(fù)制沖突。

其次，構(gòu)建當前遍歷的子文件夾的完整路徑year_folder_path，并使用os.listdir()函數(shù)獲取子文件夾中的所有文件。隨后，再遍歷剛剛得到的子文件夾中的每個文件——如果文件名包含STB，那么這個文件就是我們需要的文件；我們就構(gòu)建源文件（待復(fù)制的文件）的完整路徑source_file和目標文件（預(yù)計復(fù)制完畢的文件）的完整路徑target_file，并緊接著使用shutil.copy()函數(shù)將源文件復(fù)制到目標文件。