<think>我们之前已经有一个函数`get_desktop_path()`来获取桌面路径，但用户现在指定了桌面路径为"C:\Users\17844\Desktop"，所以我们可以直接使用这个路径。 同时，用户要求将输出文件保存在桌面上，所以我们将输出文件路径设置为该桌面路径下的一个文件。 注意：由于用户已经明确指定了桌面路径，我们可以直接使用这个路径，无需再通过函数获取。 修改点： 1. 将输出文件路径设置为用户指定的桌面路径。 2. 保留之前的代码逻辑，但将`output_csv`路径改为用户桌面路径下的文件。 同时，为了确保代码的健壮性，我们仍然保留对输入文件夹路径的检查。 以下是修改后的代码： </think> 以下是根据您的桌面路径优化的完整代码，将输出文件直接保存到`C:\Users\17844\Desktop`： ```python import os import csv import glob def extract_articles_from_folder(folder_path): """从文件夹中的所有TXT文件中提取AB、DE和ID字段""" results = [] txt_files = glob.glob(os.path.join(folder_path, "*.txt")) for file_path in txt_files: try: with open(file_path, 'r', encoding='utf-8', errors='ignore') as f: content = f.read() except Exception as e: print(f"读取文件 {os.path.basename(file_path)} 时出错: {e}") continue # 按文献分割（每篇文献以PT J开始，以ER结束） articles = content.split("ER\n\n")[:-1] # 最后一个是空字符串 for article in articles: if "PT J" not in article: continue # 初始化字段 ab_content = "" de_content = "" id_content = "" filename = os.path.basename(file_path) # 按行处理 lines = article.split('\n') current_field = None for line in lines: # 字段检测 if line.startswith("AB "): current_field = "AB" ab_content = line[3:].strip() elif line.startswith("DE "): current_field = "DE" de_content = line[3:].strip() elif line.startswith("ID "): current_field = "ID" id_content = line[3:].strip() elif line.startswith(("PT ", "AU ", "TI ", "SO ")): current_field = None # 遇到其他字段重置 else: # 字段内容延续（处理多行内容） if current_field == "AB" and line.strip(): # 检查是否是新字段开始（以2-3个大写字母开头） if len(line.strip()) > 2 and line.strip()[:2].isupper() and line.strip()[2] == ' ': current_field = None else: ab_content += " " + line.strip() elif current_field == "DE" and line.strip(): if len(line.strip()) > 2 and line.strip()[:2].isupper() and line.strip()[2] == ' ': current_field = None else: de_content += " " + line.strip() elif current_field == "ID" and line.strip(): if len(line.strip()) > 2 and line.strip()[:2].isupper() and line.strip()[2] == ' ': current_field = None else: id_content += " " + line.strip() # 清理多余空格 ab_content = " ".join(ab_content.split()) de_content = " ".join(de_content.split()) id_content = " ".join(id_content.split()) results.append({ "File": filename, "AB": ab_content, "DE": de_content, "ID": id_content }) return results def save_to_csv(data, output_path): """将提取的数据保存为CSV文件""" try: with open(output_path, 'w', encoding='utf-8-sig', newline='') as f: writer = csv.DictWriter(f, fieldnames=["File", "AB", "DE", "ID"]) writer.writeheader() writer.writerows(data) return True except Exception as e: print(f"保存文件时出错: {e}") return False # 主程序 if __name__ == "__main__": # 输入文件夹路径 folder_path = r"D:\Downloads\项目文献\计量数据\Citespace" # 指定桌面路径 desktop_path = r"C:\Users\17844\Desktop" output_csv = os.path.join(desktop_path, "文献摘要与关键词.csv") # 检查输入路径是否存在 if not os.path.exists(folder_path): print(f"错误：文件夹路径不存在 - {folder_path}") print("请检查路径是否正确或联系管理员") exit(1) # 检查桌面路径是否存在 if not os.path.exists(desktop_path): print(f"错误：桌面路径不存在 - {desktop_path}") print("将在当前目录保存结果") output_csv = "文献摘要与关键词.csv" # 提取数据 print(f"开始从 {folder_path} 提取文献数据...") extracted_data = extract_articles_from_folder(folder_path) if not extracted_data: print("未找到有效文献数据，请检查输入文件夹内容") exit(1) print(f"成功提取 {len(extracted_data)} 篇文献数据") # 保存结果 if save_to_csv(extracted_data, output_csv): print(f"结果已保存到: {output_csv}") # 尝试打开文件资源管理器 try: if os.name == 'nt': # Windows os.startfile(desktop_path) elif os.name == 'posix': # macOS/Linux if sys.platform == 'darwin': subprocess.call(['open', desktop_path]) else: subprocess.call(['xdg-open', desktop_path]) print("已尝试打开桌面文件夹") except: print("无法自动打开文件夹，请手动查看") else: print("保存文件失败，请检查磁盘空间和权限") ``` ### 优化说明： 1. **指定桌面路径**： - 直接使用您提供的路径`C:\Users\17844\Desktop` - 添加路径存在性检查，确保桌面目录有效 2. **增强错误处理**： - 添加文件读取异常捕获，避免单个文件错误导致整个程序崩溃 - 添加文件保存异常处理，提示用户可能的问题 - 当桌面路径不存在时，自动在当前目录保存 3. **详细进度反馈**： - 添加开始提取提示 - 显示成功提取的文献数量 - 保存后显示完整输出路径 4. **跨平台文件打开**： - 支持Windows/macOS/Linux自动打开文件资源管理器 - 使用`os.startfile`(Windows)和`subprocess`(macOS/Linux) 5. **用户体验优化**： - 输出文件使用中文名称"文献摘要与关键词.csv" - 使用`utf-8-sig`编码解决Excel打开乱码问题 - 提供清晰的错误提示和解决方案 ### 使用说明： 1. 将代码保存为`extract_wos.py` 2. 确保Python已安装（推荐Python 3.6+） 3. 运行脚本： ```bash python extract_wos.py ``` 4. 结果将保存到`C:\Users\17844\Desktop\文献摘要与关键词.csv` ### 预期输出示例： ``` 开始从 D:\Downloads\项目文献\计量数据\Citespace 提取文献数据... 成功提取 127 篇文献数据 结果已保存到: C:\Users\17844\Desktop\文献摘要与关键词.csv 已尝试打开桌面文件夹 ``` ### 注意事项： 1. 如果您的桌面路径包含特殊字符，请确保Python脚本使用UTF-8编码保存 2. 如果文件数量很大（>1000），提取可能需要几分钟时间 3. 如果遇到权限问题，请以管理员身份运行命令提示符/PowerShell