Python實現(xiàn)批量文件整理的示例代碼

更新時間：2021年12月08日 10:34:50 作者：幸福的達哥

批量文件整理一直是日常工作中令人頭疼的事，使用 Python 進行大批量文件整理，可以大大提升工作效率。本文主要介紹了利用Python實現(xiàn)文件的重命名和刪除，感興趣的小伙伴可以關(guān)注一下

引言?

批量文件整理一直是日常工作中令人頭疼的事，使用 Python 進行大批量文件整理，可以大大提升工作效率。下面來介紹幾種批量文件整理的小技巧。

一、準(zhǔn)備工作

為了用于實驗，我們使用代碼生成 200 個 txt 文件，代碼如下。

for i in range(0, 200):
    file_name = f'file_{i}.txt'
    f = open(f'./file/{file_name}', mode='w')
    f.close()

運行結(jié)果：

二、制作 excel 的文件清單

1、思路

獲取文件名，并將序號和文件名寫入 excel。

2、openpyxl 安裝

本文使用 openpyxl 庫進行 excel 操作，使用 pip 進行安裝。

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple openpyxl

3、代碼

from openpyxl import load_workbook
import os
 
# 獲取file路徑所有txt文件
def find_txt(path):
    filenames = []
    filename_listdir = os.listdir(path)
    for filename in filename_listdir:
        if filename.find('txt') != -1:
            filenames.append(filename)
    return filenames
 
# 生成文件清單
def add_data(excel_path, filenames):
    # 判斷excel文件是否存在
    if os.path.exists(excel_path) is False:
				print(excel_path + ' 文件不存在，請重試')
				exit()
    excel_file = load_workbook(excel_path)  # 打開excel文件
    excel_sheet = excel_file['Sheet1']  # 選擇Sheet1
    # 增加表頭
    excel_sheet.cell(row=1, column=1, value='序號')  # 序號
    excel_sheet.cell(row=1, column=2, value='文件名')  # 文件名
    # 添加文件名
    count = 1
    for i in filenames[0:]:
        count = count + 1
        excel_sheet.cell(row=count, column=1, value=count - 1)  # 序號
        excel_sheet.cell(row=count, column=2, value=i)  # 文件名
    excel_file.save(excel_path)
 
# 文件所在文件夾
file_path = './file'
# 【文件清單.xlsx】路徑
excel_path = os.getcwd() + '/file/文件清單.xlsx'
filenames = find_txt(file_path)
print(filenames)
add_data(excel_path, filenames)
print('Success！')

運行結(jié)果：

三、文件的批量重命名

1、文件名批量增加序號

在文件整理統(tǒng)計時，經(jīng)常要給文件名添加序號，在上面保存文件清單的過程，我們可以看到腳本的默認(rèn)排序保存是按首位數(shù)字排序，在此我們可以通過補零來保證按數(shù)字大小排序。

1）重命名函數(shù)：

os.rename(name, new_name)

2）代碼

import os
 
path = os.getcwd() + '\\file'
filenames = os.listdir(path)
 
a = 1
for filename in filenames:
    if filename.find('txt') != -1:
        old_dir = f'{path}\\{filename}'
        if a < 10:
            new_dir = f'{path}\\00{a}-{filename}'
        elif a < 100:
            new_dir = f'{path}\\0{a}-{filename}'
        else:
            new_dir = f'{path}\\{a}-{filename}'
        os.rename(old_dir, new_dir)
        a = a + 1

3）運行結(jié)果

2、文件名批量重命名成指定文件名

我們在制作文件清單后，可以使用excel進行便捷的文件名統(tǒng)一調(diào)整（如去空格，增加頭尾字段等）。我們可以在excel中制作好修改后的文件名后，再批量將文件名重命名。

1）實驗?zāi)繕?biāo)

如圖：我們嘗試將文件名改成新文件名（使用 excel 的將 file 替換成filename ）。

2）代碼

from openpyxl import load_workbook
import os
 
# 獲取file路徑文件名
def find_txt(path):
    filenames = []
    filename_listdir = os.listdir(path)
    for filename in filename_listdir:
        if filename.find('txt') != -1:
            filenames.append(filename)
    return filenames
 
# 批量重命名
def change_file_name(file_path, excel_path, filenames):
    data = load_workbook(excel_path)
    sheet = data['Sheet1']
    for i in range(1, sheet.max_row + 1)[1:]:
        for filename in filenames:
            if filename == sheet.cell(i, 2).value:
                old_dir = os.path.join(file_path, filename)
                new_dir = os.path.join(file_path, sheet.cell(i, 3).value)
                os.rename(old_dir, new_dir)
            else:
                pass
# 文件所在文件夾
file_path = './file'
# 【文件清單.xlsx】路徑
excel_path = os.getcwd() + '/file/文件清單.xlsx'
filenames = find_txt(file_path)
print(filenames)
change_file_name(file_path, excel_path, filenames)
print('Success！')

運行效果：

四、文件的批量刪除

1、實驗?zāi)繕?biāo)

我們在excel對文件名進行篩選后，將是否刪除列為1的文件刪除，保留為0的文件（如圖）。

2、代碼

from openpyxl import load_workbook
import os
 
# 獲取file路徑文件名
def find_txt(path):
    filenames = []
    filename_listdir = os.listdir(path)
    for filename in filename_listdir:
        if filename.find('txt') != -1:
            filenames.append(filename)
    return filenames
 
# 批量重命名
def change_file_name(file_path, excel_path, filenames):
    data = load_workbook(excel_path)
    sheet = data['Sheet1']
    for i in range(1, sheet.max_row + 1)[1:]:
        for filename in filenames:
            if filename == sheet.cell(i, 2).value:
                file_dir = os.path.join(file_path, filename)
                delete_flag = sheet.cell(i, 3).value
                if delete_flag:
                    os.remove(file_dir)
            else:
                pass
# 文件所在文件夾
file_path = './file'
# 【文件清單.xlsx】路徑
excel_path = os.getcwd() + '/file/文件清單.xlsx'
filenames = find_txt(file_path)
print(filenames)
change_file_name(file_path, excel_path, filenames)
print('Success！')