### 如何根据类型或内容对Python文件进行分类 对于 Python 文件的分类可以基于多种标准执行，比如按照文件的内容特征、函数定义的数量或是特定关键字的存在与否等。下面提供一种方法来依据文件内是否存在某些预设的关键字来进行分类。 #### 准备工作 为了能够顺利地完成这项任务，需要先安装 `os` 和 `re` 库用于操作文件路径以及正则表达式的匹配： ```python import os import re ``` #### 定义关键词列表 创建一个包含不同类别对应关键字的字典，这些关键字将帮助区分各个类别的 Python 脚本: ```python categories_keywords = { "web": ["flask", "django"], "data_analysis": ["pandas", "numpy"], "machine_learning": ["tensorflow", "keras"] } ``` #### 遍历目标目录下的所有 .py 文件 通过遍历指定的目标文件夹内的 `.py` 文件，并读取它们的内容以便后续分析： ```python def traverse_files(directory): python_files = [] for root, dirs, files in os.walk(directory): for file in files: if file.endswith(".py"): full_path = os.path.join(root, file) with open(full_path, 'r', encoding='utf-8') as f: content = f.read() python_files.append((full_path, content)) return python_files ``` #### 进行分类判断并将结果保存到相应的子文件夹中 针对每一个被读取出来的 Python 文件，检查其内容是否含有任何已知的关键字组合；如果存在，则将其移动至对应的分类下新建立的子文件夹里： ```python def classify_and_save(python_files, output_dir="classified_scripts"): categorized_counts = {category: 0 for category in categories_keywords} try: os.makedirs(output_dir, exist_ok=True) for path, content in python_files: matched_category = None for cat, keywords in categories_keywords.items(): pattern = "|".join(keywords) match_result = re.search(pattern, content.lower()) if match_result is not None: matched_category = cat break destination_folder = os.path.join(output_dir, str(matched_category)) if matched_category else os.path.join(output_dir, "uncategorized") os.makedirs(destination_folder, exist_ok=True) new_file_name = os.path.basename(path).replace('.py', '_copy.py') shutil.copyfile(path, os.path.join(destination_folder, new_file_name)) if matched_category: categorized_counts[matched_category] += 1 print(f"Categorization completed! Files have been organized into the following groups:\n{categorized_counts}") except Exception as e: print(e) if __name__ == "__main__": target_directory = "./your_target_directory_here" all_python_files = traverse_files(target_directory) classify_and_save(all_python_files) ``` 此脚本会扫描给定目录及其子目录中的所有 Python 文件，并尝试根据预先设定好的关键字集对其进行分类。最终的结果会被复制并放置在一个新的名为 `classified_scripts` 的根级文件夹之下，其中每个子文件夹代表一类 Python 程序[^1]^。