快捷導(dǎo)航

Python批量查找包含多個關(guān)鍵詞的PDF文件

更新時間：2024年11月01日 08:34:59 作者：PythonFun

在信息爆炸的時代,數(shù)據(jù)管理變得愈發(fā)重要,本文主要為大家介紹了如何通過Python批量查找包含多個關(guān)鍵詞的PDF文件,希望對大家有所幫助

在信息爆炸的時代，數(shù)據(jù)管理變得愈發(fā)重要。U盤作為一種便攜式存儲設(shè)備，常常承載著我們大量的個人和工作數(shù)據(jù)。然而，隨著文件數(shù)量的增加，在U盤中快速找到特定文件常常成為一個令人頭疼的難題。我們通?？梢圆捎胑verything來快速查找我們想要的文件，但是everything只有查找功能，并沒有復(fù)制功能，同時不能進行批量的查找，所以這時就可能要使用到萬能的Python工具了，Python中的os標準模塊可以遍歷，查找文件，再用shutil來文件拷貝到指定位置。

一、查找包含單一關(guān)鍵詞的文件

比如我們要查找U盤中包括："翻譯", "國際", "英語", "國別" 任一關(guān)鍵詞的pdf文件，找到后復(fù)制到查找文件這一個目錄下，如果找不到就重新建立查找文件這個文件夾。代碼如下：

import os
import shutil
 
# 定義要查找的關(guān)鍵詞
keywords = ["翻譯", "國際", "英語", "國別"]
 
# 定義要遍歷的目錄和目標目錄
source_directory = "你的源目錄路徑"  # 替換為你的U盤路徑，如果是當前目錄下可以直接填寫為"."
target_directory = os.path.join(source_directory, "查找文件")
 
# 如果目標目錄不存在，則創(chuàng)建它
if not os.path.exists(target_directory):
    os.makedirs(target_directory)
 
# 遍歷源目錄下的所有文件和文件夾
for root, dirs, files in os.walk(source_directory):
    for file in files:
        # 檢查文件名是否包含任何關(guān)鍵詞，并且是PDF文件
        if any(keyword in file for keyword in keywords) and file.endswith('.pdf'):
            source_file_path = os.path.join(root, file)
            target_file_path = os.path.join(target_directory, file)
            # 復(fù)制文件到目標目錄
            shutil.copy(source_file_path, target_file_path)
            print(f"已復(fù)制: {source_file_path} 到 {target_file_path}")
 
print("文件查找和復(fù)制完成！")

使用時，將 source_directory 替換為你要遍歷的目錄的路徑。

運行代碼，它會遍歷指定的目錄，查找包含關(guān)鍵詞的PDF文件并復(fù)制到“查找文件”文件夾中。

確保在你的Python環(huán)境中安裝了所需的模塊（如 os 和 shutil），這些模塊通常是Python標準庫的一部分，無需額外安裝。

二、查找多關(guān)鍵詞同時出現(xiàn)的文件

查找同時包含“翻譯”和“碩”兩個關(guān)鍵詞的PDF文件名，并將其復(fù)制到“查找文件”文件夾中呢？

import os
import shutil
 
# 定義要查找的關(guān)鍵詞
keywords = ["翻譯", "碩"]
 
# 定義要遍歷的目錄和目標目錄
source_directory = "你的源目錄路徑"  # 替換為你的源目錄路徑
target_directory = os.path.join(source_directory, "查找文件")
 
# 如果目標目錄不存在，則創(chuàng)建它
if not os.path.exists(target_directory):
    os.makedirs(target_directory)
 
# 遍歷源目錄下的所有文件和文件夾
for root, dirs, files in os.walk(source_directory):
    for file in files:
        # 檢查文件名是否同時包含所有關(guān)鍵詞，并且是PDF文件
        if all(keyword in file for keyword in keywords) and file.endswith('.pdf'):
            source_file_path = os.path.join(root, file)
            target_file_path = os.path.join(target_directory, file)
            # 復(fù)制文件到目標目錄
            shutil.copy(source_file_path, target_file_path)
            print(f"已復(fù)制: {source_file_path} 到 {target_file_path}")
 
print("文件查找和復(fù)制完成！")

使用說明：

將 source_directory 替換為你要遍歷的目錄的路徑。

運行代碼，它會查找同時包含“翻譯”和“碩”的PDF文件并將其復(fù)制到“查找文件”文件夾中。

三、把以上兩種功能合二為一

設(shè)置選項，當用戶輸入不同的選項就進行不同的操作。

import os
import shutil
 
def find_files_any(source_directory, keywords):
    target_directory = os.path.join(source_directory, "查找文件_any")
    if not os.path.exists(target_directory):
        os.makedirs(target_directory)
        
    for root, dirs, files in os.walk(source_directory):
        for file in files:
            if any(keyword in file for keyword in keywords):
                print(f"找到任一關(guān)鍵詞文件: {file}")
 
def find_files_all(source_directory, keywords):
    target_directory = os.path.join(source_directory, "查找文件_all")
    if not os.path.exists(target_directory):
        os.makedirs(target_directory)
        
    for root, dirs, files in os.walk(source_directory):
        for file in files:
            if all(keyword in file for keyword in keywords):
                source_file_path = os.path.join(root, file)
                target_file_path = os.path.join(target_directory, file)
                shutil.copy(source_file_path, target_file_path)
                print(f"已復(fù)制: {source_file_path} 到 {target_file_path}")
 
def main():
    keywords_any = ["翻譯", "國際"]
    keywords_all = ["翻譯", "碩"]
    
    print("請選擇查找選項：")
    print("1. 查找任一關(guān)鍵詞")
    print("2. 查找同時關(guān)鍵詞")
    
    choice = input("請輸入選項 (1 或 2): ")
    
    source_directory = input("請輸入你的U盤路徑: ")
    
    if choice == "1":
        find_files_any(source_directory, keywords_any)
    elif choice == "2":
        find_files_all(source_directory, keywords_all)
    else:
        print("無效選項，請重新運行程序。")
 
    print("文件查找完成！")
 
if __name__ == "__main__":
    main()

顯示情況如下：

顯示結(jié)果

四、采用裝飾器法來寫

為了使我們的代碼更pythonic，我們可以設(shè)置一下裝飾器，這樣可以為我們設(shè)置的函數(shù)添加新的功能。

import os
import shutil
 
def choice_decorator(func):
    def wrapper(keywords):
        print("請選擇查找選項：")
        print("1. 查找任一關(guān)鍵詞")
        print("2. 查找同時關(guān)鍵詞")
        
        choice = input("請輸入選項 (1 或 2): ")
        
        if choice not in ["1", "2"]:
            print("無效選項，請重新運行程序。")
            return
        
        source_directory = input("請輸入你的U盤路徑: ")
        
        if choice == "1":
            return func(source_directory, keywords[0])  # 傳遞任一關(guān)鍵詞
        elif choice == "2":
            return func(source_directory, keywords[1])  # 傳遞同時關(guān)鍵詞
    
    return wrapper
 
@choice_decorator
def find_files(source_directory, keywords):
    target_directory = os.path.join(source_directory, f"查找文件_{keywords[0]}")
    if not os.path.exists(target_directory):
        os.makedirs(target_directory)
    
    for root, dirs, files in os.walk(source_directory):
        for file in files:
            if all(keyword in file for keyword in keywords):
                source_file_path = os.path.join(root, file)
                target_file_path = os.path.join(target_directory, file)
                shutil.copy(source_file_path, target_file_path)
                print(f"已復(fù)制: {source_file_path} 到 {target_file_path}")
 
def main():
    keywords_any = ["翻譯", "國際"]
    keywords_all = ["翻譯", "碩"]
    
    # 將關(guān)鍵詞組合放在一個列表中，以便裝飾器使用
    keywords = [keywords_any, keywords_all]
    
    find_files(keywords)
 
    print("文件查找完成！")
 
if __name__ == "__main__":
    main()