1. 从脚本小子到计算化学家为什么你需要掌握Schrödinger Python API如果你和我一样在计算化学或者药物设计的领域里摸爬滚打了一段时间大概率会经历这样一个阶段每天在Maestro的图形界面里点来点去重复着“导入结构-设置参数-提交任务-等待结果-导出数据”的循环。刚开始觉得图形界面真方便所见即所得。但项目做多了特别是需要处理几十上百个分子、进行高通量筛选或者复杂工作流自动化时这种手动操作就变成了噩梦。效率低下不说还容易出错一个手滑点错参数可能一晚上的计算就白跑了。这时候Schrödinger Python API通常我们简称为PySchrödinger就成了从“操作工”进阶为“工程师”的关键钥匙。它不是一个独立的软件而是Schrödinger Suite一个集成了分子模拟、药物设计等强大工具的商业软件套件提供给我们的一个编程接口。简单来说它允许你用Python代码去“指挥”Maestro、Glide、Prime、Desmond这些强大的计算引擎把繁琐的图形界面操作变成一行行可重复、可追溯、可批处理的脚本。我最初接触它是因为一个激进的虚拟筛选项目需要对接上万个化合物到同一个蛋白的多个构象。手动不可能。正是PySchrödinger让我在一周内完成了原本需要数月手动操作的任务并且所有参数、所有步骤都记录在脚本里可随时复现和审查。这不仅仅是效率的提升更是工作方式的革命——从依赖直觉和手工转向基于代码的、可工程化的研究流程。所以无论你是计算化学的研究生还是药物发现一线的科学家如果你已经厌倦了重复劳动渴望将创造力从机械操作中解放出来投入到更核心的科学问题中去那么这个系列就是为你准备的。我们将从“为什么需要它”开始一步步拆解这个强大工具的核心概念、基础操作和实战技巧最终让你能亲手打造属于自己的自动化研究管线。2. PySchrödinger生态全景它到底是什么能做什么在深入代码之前我们必须先搞清楚PySchrödinger在整个Schrödinger生态中的位置以及它的能力边界。这有助于我们建立正确的心理预期知道该用它解决什么问题以及哪些问题它可能不擅长。2.1 核心定位连接Python与计算引擎的桥梁首先要破除一个常见的误解PySchrödinger并不是一个用来“重新发明轮子”的独立计算库。它不提供从头实现分子力学力场、量子化学计算或者分子对接算法的新函数。相反它的核心价值在于“封装”和“驱动”。你可以把它想象成一个超级遥控器。Schrödinger Suite里的各个模块Glide对接、Prime优化、Desmond动力学等就像是一台台功能强大但操作复杂的专业设备比如高级相机、录音棚调音台。PySchrödinger就是这个统一的遥控器它定义了标准的“按键”函数和方法让你可以用Python程序来按下这些按键从而远程、精确地控制这些设备工作并获取它们产出的结果。这种设计带来了几个根本性优势自动化将图形界面的点击流程转化为脚本实现7x24小时无人值守的批量计算。可重复性脚本即记录。任何分析流程都可以被精确复现这对于科学研究至关重要。集成性可以轻松地将Schrödinger的计算能力嵌入到你用Python构建的更大数据分析流水线中与NumPy、Pandas、Scikit-learn等科学生态系统无缝衔接。灵活性可以编写复杂的逻辑条件判断、循环、自定义函数来控制计算流程这是图形界面难以实现的。2.2 核心模块与能力地图PySchrödinger的API是模块化的大致对应着Schrödinger Suite中的主要应用程序。了解这些模块你就知道了它的“武器库”里有什么。schrodinger.structure(结构模块)这是最基础、使用最频繁的模块。它提供了Structure对象用于在内存中表示和操作分子结构原子、键、坐标、属性。你可以用它来读取/写入各种文件格式.mae,.sdf,.pdb,.mol2遍历原子、修改坐标、计算简单的几何性质距离、角度、二面角、处理子结构等。几乎所有涉及分子数据输入输出的操作都从这里开始。schrodinger.application(应用模块)这是驱动计算引擎的核心。其下又有众多子模块schrodinger.application.glide用于运行Glide分子对接。你可以用代码设置搜索框、精度、打分函数等所有在Glide面板上能看到的参数。schrodinger.application.prime用于运行Prime的蛋白质结构优化、侧链预测、能量计算等。schrodinger.application.desmond用于设置和提交Desmond分子动力学模拟。schrodinger.application.ligprep用于运行LigPrep进行配体预处理加氢、生成互变异构体、电离状态等。schrodinger.application.jaguar用于运行Jaguar量子化学计算。schrodinger.job(任务管理模块)用于提交和管理计算任务。它封装了作业提交到本地机器、局域网集群或云平台上的细节。你可以用它来启动一个Glide作业并监控其运行状态等待、运行、完成、失败。schrodinger.analysis(分析模块)包含一些用于分析计算结果的工具例如分析动力学轨迹、计算结合自由能MM/GBSA等。不过很多深入的分析仍需结合第三方库如MDTraj、MDAnalysis或自定义脚本。schrodinger.utils(工具模块)提供一些通用工具函数如文件操作、日志记录等。注意PySchrödinger的官方文档有时更新不及时且某些高级功能可能缺乏示例。最可靠的学习方式往往是结合官方文档、查看Schrödinger安装目录下的示例脚本通常在$SCHRODINGER/utilities/python/或类似路径以及在实际项目中摸索。2.3 工作模式两种主要的交互方式PySchrödinger主要支持两种工作模式适用于不同场景“库”模式 (In-Process)在你的Python脚本中直接导入schrodinger模块调用其函数。这些函数会在当前Python进程内调用Schrödinger的编译库执行计算。这种方式速度快适合轻量级、不需要复杂任务管理的操作比如读取一个结构文件、进行简单的几何变换。# 示例库模式读取文件 from schrodinger.structure import StructureReader with StructureReader(ligands.mae) as reader: for st in reader: print(st.title, st.atom_total)“作业”模式 (Out-of-Process)通过schrodinger.job模块创建作业对象并提交。计算任务会作为一个独立的进程或在远程服务器上运行。这是运行耗时计算如对接、动力学的标准方式。你的脚本提交作业后可以继续执行其他逻辑或者等待作业完成。# 示例作业模式提交Glide对接 from schrodinger.application.glide import glide from schrodinger.job import jobcontrol # 创建Glide作业对象并配置参数 glide_job glide.Glide() glide_job.setInputFile(receptor.mae) glide_job.setLigandFile(ligands.mae) glide_job.setPrecision(glide.Precision.SP) # 标准精度 # 提交作业 job_id jobcontrol.submit(glide_job) print(fJob submitted with ID: {job_id}) # 可选项等待作业完成 jobcontrol.wait_for_job(job_id)理解这两种模式的区别至关重要它决定了你脚本的架构。大部分生产环境下的自动化流程都是两种模式混合使用用“库”模式准备输入文件用“作业”模式提交重型计算再用“库”模式分析输出结果。3. 环境搭建与“Hello World”迈出第一步理论说再多不如动手跑一行代码。搭建PySchrödinger的开发环境有其特殊性因为它深度依赖于Schrödinger Suite的安装。3.1 安装前提Schrödinger Suite是基石最重要的一点你必须先拥有一个正确安装并激活许可的Schrödinger Suite。PySchrödinger是随套件一起安装的不是一个可以通过pip install schrodinger获取的独立包。通常安装完成后Schrödinger会提供一个专门配置好的Python解释器。找到它Linux/macOS通常在$SCHRODINGER/utilities/python目录下。你可以通过$SCHRODINGER/run python命令直接启动这个解释器。Windows通常在Schrödinger的安装目录下例如C:\Schrodinger2024-1\utilities\python.exe。也可以在开始菜单的Schrödinger程序组里找到名为“Schrödinger Python”的快捷方式。为什么必须用它因为这个Python解释器已经预编译链接了所有Schrödinger的底层C/C库并正确设置了环境变量如SCHRODINGER。如果你使用系统自带的Python或Anaconda环境直接import schrodinger十有八九会失败报错找不到模块或库文件。3.2 配置你的开发环境以VS Code为例虽然可以用命令行直接操作但使用一个集成开发环境IDE会极大提升效率。这里以VS Code为例展示如何配置。创建项目目录为你的一系列脚本创建一个干净的文件夹例如my_pyschrodinger_projects。在VS Code中打开该文件夹。配置Python解释器按下CtrlShiftP(Windows/Linux) 或CmdShiftP(macOS)输入“Python: Select Interpreter”。选择“Enter interpreter path”然后浏览并选中Schrödinger提供的Python解释器路径例如C:\Schrodinger2024-1\utilities\python.exe。选择后VS Code右下角会显示当前使用的解释器。创建虚拟环境可选但推荐虽然Schrödinger的Python环境是独立的但为了安装一些额外的数据分析包如pandas, matplotlib最好在其基础上创建一个虚拟环境。# 在项目目录下打开终端使用Schrödinger Python # 假设Schrödinger Python路径已加入系统PATH名为‘schrodinger_python’ schrodinger_python -m venv venv # 激活虚拟环境 # Windows: .\venv\Scripts\activate # Linux/macOS: source venv/bin/activate # 在激活的虚拟环境中pip安装的包会独立存放 pip install pandas matplotlib jupyter然后在VS Code中再次选择解释器这次选择虚拟环境下的python.exe例如my_pyschrodinger_projects\venv\Scripts\python.exe。3.3 你的第一个脚本读取并查看分子让我们写一个最简单的脚本来验证环境并建立直观感受。创建一个新文件命名为hello_pyschrodinger.py。#!/usr/bin/env python 第一个PySchr?dinger脚本读取分子文件并打印基本信息。 确保你有一个可用的.mae或.sdf文件或者使用Schr?dinger自带的示例文件。 # 1. 导入核心模块 from schrodinger.structure import StructureReader, StructureWriter import sys def main(input_file): 主函数读取分子文件并打印信息。 :param input_file: 输入的分子文件路径如.mae, .sdf格式 # 2. 使用StructureReader读取文件 # StructureReader是一个迭代器可以高效地处理包含多个分子的文件 print(f正在读取文件: {input_file}) try: # ‘with‘语句确保文件被正确关闭 with StructureReader(input_file) as reader: # 3. 遍历文件中的每一个分子结构 for i, st in enumerate(reader): print(f\n--- 分子 {i1} ---) print(f 标题: {st.title}) print(f 原子总数: {st.atom_total}) print(f 化学式: {st.formula}) # 4. 访问并打印前几个原子的信息避免输出过长 print(f 前5个原子信息:) for atom in st.atom[:5]: # st.atom是一个原子列表 # atom对象包含索引、元素符号、坐标、电荷等属性 print(f 原子索引 {atom.index}: 元素 {atom.element}, f坐标 ({atom.x:.3f}, {atom.y:.3f}, {atom.z:.3f})) # 5. 演示一个简单操作计算分子量近似 # 注意这是基于原子量的简单加和未考虑同位素 mol_weight sum(atom.atom_weight for atom in st.atom) print(f 近似分子量: {mol_weight:.2f} g/mol) except FileNotFoundError: print(f错误找不到文件 {input_file}) sys.exit(1) except Exception as e: print(f读取文件时发生未知错误: {e}) sys.exit(1) if __name__ __main__: # 检查命令行参数 if len(sys.argv) ! 2: print(用法: python hello_pyschrodinger.py 分子文件路径) print(示例: python hello_pyschrodinger.py example.mae) sys.exit(1) input_file_path sys.argv[1] main(input_file_path)如何运行准备一个分子文件。如果你没有可以使用Maestro创建一个简单的分子如丙烷并保存为.mae格式或者到Schrödinger安装目录下寻找示例文件例如在$SCHRODINGER/mmshare-vX.Y/example_files/里。在终端中确保你的Python环境已激活如果用了虚拟环境然后运行python hello_pyschrodinger.py 你的分子文件.mae如果一切正常你将看到终端打印出该分子的基本信息。这段代码揭示了什么Structure对象是核心st变量代表一个完整的分子它是所有操作的起点。原子是可迭代和可索引的st.atom是一个列表你可以通过索引st.atom[0]或循环来访问每一个原子。属性访问非常直观像st.title,atom.element,atom.x这样直接访问符合Python的简洁哲学。文件读取是流式的StructureReader即使面对包含成千上万个分子的超大文件也能高效地一次处理一个不会全部加载到内存。这个简单的脚本已经包含了PySchrödinger最基础、最常用的模式。运行成功意味着你的环境已经就绪我们可以开始探索更强大的功能了。4. 深入Structure对象分子数据的“瑞士军刀”上一节我们接触了Structure对象。如果说PySchrödinger是一套乐高那Structure就是最基础、最重要的那块积木。几乎所有工作都始于读取一个结构终于写入或分析一个结构。因此彻底理解Structure对象至关重要。4.1 结构属性不仅仅是原子坐标一个Structure对象包含的信息远不止原子列表。它由多个“层面”的数据构成原子层面 (Atom Level)这是最核心的数据。每个原子是一个_Atom对象通常我们直接叫它atom其关键属性包括index: 原子在结构中的唯一整数索引从1开始。element: 元素符号字符串如 ‘C‘ ‘N‘ ‘O‘。x, y, z: 三维空间坐标浮点数。charge: 部分电荷浮点数。atom_type: 力场原子类型字符串。pdbname: PDB格式中的原子名称如 ‘ CA ‘。resnum: 残基序号整数。chain: 链标识符字符串如 ‘A‘。inscode: 插入码字符串用于处理PDB中残基序号不连续的情况。结构层面 (Structure Level)整个分子或体系的全局属性。title: 结构的标题通常从文件名或注释行读取。property: 这是一个类似字典的接口用于访问和修改存储在结构文件中的各种属性。这是Structure对象最强大也最容易让人困惑的特性之一。分子属性如r_mmod_Potential_Energy(分子力学势能)s_m_title(标题)i_m_ct_format(格式版本)等。原子属性每个原子也可以有独立的属性通过atom.property访问如r_j_Atm_polarizability(原子极化率)。这些属性键key通常是schrodinger内部定义的字符串很多以r_(实数),i_(整数),s_(字符串),b_(布尔值) 开头表示数据类型。化学信息层面bond: 键的列表。每个键对象连接两个原子索引并包含键级单键、双键等信息。formula: 化学式字符串。4.2 属性Property系统详解与实战property系统是Schrödinger文件格式如.mae存储额外数据的核心机制。理解它你才能自由地读写计算产生的各种数据如对接打分、能量值、RMSD等。如何查看一个结构有哪些属性from schrodinger.structure import StructureReader with StructureReader(‘complex.mae‘) as reader: st next(reader) # 读取第一个结构 # 获取所有结构级别的属性名 prop_names st.property.keys() print(结构属性列表:) for name in prop_names: # 获取属性值及其数据类型 value st.property[name] print(f {name}: {value} (类型: {type(value).__name__})) # 查看第一个原子的属性如果有的话 if st.atom_total 0: atom st.atom[0] atom_prop_names atom.property.keys() print(f\n第一个原子的属性:) for name in atom_prop_names[:5]: # 只打印前5个避免过多 print(f {name}: {atom.property[name]})如何读写属性# 写入一个新的结构属性 st.property[‘s_my_custom_notes‘] ‘This is a manually added note.‘ # 写入一个新的原子属性为每个原子添加 for atom in st.atom: atom.property[‘r_my_custom_charge‘] 0.0 # 假设我们自定义一个电荷 # 读取Glide对接结果中常见的打分属性 if ‘r_i_docking_score‘ in st.property: docking_score st.property[‘r_i_docking_score‘] print(fDocking Score: {docking_score})踩坑点1属性名的“黑盒”性。很多属性名如r_i_docking_score是Schrödinger内部定义的文档可能不全。最可靠的方法是先用一个已知包含所需数据的文件比如一个Glide对接输出文件运行上面的查看代码找到确切的属性名然后在你的脚本中使用它。踩坑点2属性类型匹配。当你写入属性时键的前缀最好与值的类型匹配s_对应字符串i_对应整数r_对应浮点数b_对应布尔值虽然不匹配有时也能工作但可能在其他Schrödinger工具中引发问题。4.3 常用结构操作选择、修改与遍历掌握了数据存取下一步就是操作。原子选择与索引# 通过索引访问原子索引从1开始 atom_5 st.atom[5] # 获取第6个原子因为索引5对应第6个 # 遍历所有原子 for atom in st.atom: if atom.element ‘O‘: print(f找到氧原子索引: {atom.index}) # 使用列表推导式进行筛选 carbon_atoms [atom for atom in st.atom if atom.element ‘C‘] print(f碳原子数量: {len(carbon_atoms)})修改结构# 修改原子坐标例如将第一个原子移动到原点 if st.atom_total 0: st.atom[0].x 0.0 st.atom[0].y 0.0 st.atom[0].z 0.0 # 添加氢原子调用Schr?dinger的加氢算法 # 注意这需要导入特定模块且可能依赖力场 from schrodinger.structutils import build build.add_hydrogens(st) # 删除水分子示例删除链名为‘W‘的残基 # 注意直接删除原子会改变后续原子的索引需要小心处理或从后往前删 atoms_to_delete [] for atom in st.atom: if atom.chain ‘W‘: # 假设水分子在链W atoms_to_delete.append(atom.index) # 按索引从大到小删除避免索引错乱 for index in sorted(atoms_to_delete, reverseTrue): st.deleteAtom(index)几何计算from schrodinger.structutils import measure # 计算两个原子间的距离假设原子索引1和10 if st.atom_total 10: dist measure.measure_distance(st, 1, 10) print(f原子1和10之间的距离: {dist:.3f} ?) # 计算三个原子形成的角度原子索引1510 angle measure.measure_angle(st, 1, 5, 10) print(f角度(1-5-10): {angle:.2f} 度) # 计算四个原子形成的二面角扭转角 dihedral measure.measure_dihedral(st, 1, 5, 10, 15) print(f二面角(1-5-10-15): {dihedral:.2f} 度)写入文件from schrodinger.structure import StructureWriter # 写入单个结构到新文件 with StructureWriter(‘modified_structure.mae‘) as writer: writer.append(st) # 将当前结构对象写入文件 # 批量写入多个结构例如筛选后的结果 good_structures [] # 假设这是一个经过筛选的Structure对象列表 with StructureWriter(‘filtered_results.mae‘) as writer: for good_st in good_structures: writer.append(good_st)通过熟练运用Structure对象及其相关工具structutils你就能完成分子数据的预处理、后处理、分析和转换等绝大部分日常工作。这是构建任何自动化流程的基石。5. 第一个实战案例批量分子文件格式转换与信息提取现在让我们把前面学到的知识组合起来解决一个实际工作中经常遇到的问题批量处理一堆分子文件将它们从一种格式转换成另一种格式并同时提取一些关键信息保存到表格中。假设你从公共数据库下载了1000个化合物的.sdf文件你需要将它们全部转换成Maestro格式.mae因为后续的Glide对接需要这个格式。从每个分子中提取一些基本信息比如分子量、重原子数、可旋转键数并保存到一个CSV文件中方便快速筛选。我们将一步步构建这个脚本。5.1 设计脚本逻辑输入一个包含多个.sdf文件的目录。处理 a. 遍历目录中的所有.sdf文件。 b. 对于每个文件使用StructureReader读取其中的所有分子。 c. 对每个分子结构st i. 计算所需属性分子量、重原子数、可旋转键数。 ii. 将结构写入新的.mae文件可以每个分子一个文件或合并成一个文件。 iii. 将提取的属性信息存储到一个列表里。输出 a. 转换后的.mae文件。 b. 一个CSV文件包含文件名、分子标题、以及计算出的各项属性。5.2 代码实现创建一个新文件命名为batch_sdf_to_mae.py。#!/usr/bin/env python 批量SDF转MAE并提取分子信息脚本。 功能将指定文件夹内所有.sdf文件转换为.mae文件并生成包含分子信息的CSV报告。 import os import sys import glob from schrodinger.structure import StructureReader, StructureWriter from schrodinger.structutils import analyze import pandas as pd def calculate_molecular_weight(st): 计算近似分子量基于原子量加和。 :param st: Structure对象 :return: 分子量 (float) return sum(atom.atom_weight for atom in st.atom) def count_heavy_atoms(st): 计算重原子非氢原子数量。 :param st: Structure对象 :return: 重原子数 (int) return sum(1 for atom in st.atom if atom.element ! ‘H‘) def count_rotatable_bonds(st): 计算可旋转键的数量粗略估计忽略环内键和末端键。 使用Schr?dinger内置的analyze模块。 :param st: Structure对象 :return: 可旋转键数 (int) # evaluate_rotatable_bonds返回一个元组第一个元素是数量 return analyze.evaluate_rotatable_bonds(st)[0] def process_sdf_file(sdf_path, output_mae_dir, output_csv_data): 处理单个SDF文件。 :param sdf_path: 输入的SDF文件路径 :param output_mae_dir: 输出MAE文件的目录 :param output_csv_data: 用于存储CSV数据的列表 base_name os.path.splitext(os.path.basename(sdf_path))[0] output_mae_path os.path.join(output_mae_dir, f{base_name}.mae) print(f处理文件: {sdf_path}) try: with StructureReader(sdf_path) as reader, \ StructureWriter(output_mae_path) as writer: mol_count_in_file 0 for st in reader: mol_count_in_file 1 # 为每个分子生成一个唯一的标识如果文件中有多个分子 if mol_count_in_file 1: mol_title_for_csv f{base_name}_mol{mol_count_in_file} # 在结构标题中也加上标识避免重复 st.title f{st.title}_{mol_count_in_file} if st.title else mol_title_for_csv else: mol_title_for_csv base_name # 1. 计算分子属性 mol_weight calculate_molecular_weight(st) heavy_atoms count_heavy_atoms(st) rot_bonds count_rotatable_bonds(st) # 2. 将信息添加到CSV数据列表 output_csv_data.append({ ‘Source_SDF‘: os.path.basename(sdf_path), ‘Molecule_Title‘: st.title, ‘Molecular_Weight‘: round(mol_weight, 2), ‘Heavy_Atom_Count‘: heavy_atoms, ‘Rotatable_Bond_Count‘: rot_bonds, ‘Output_MAE‘: os.path.basename(output_mae_path) }) # 3. 将结构写入MAE文件 # 注意这里将所有分子写入同一个MAE文件。如果希望每个分子单独文件需调整逻辑。 writer.append(st) print(f 完成。发现 {mol_count_in_file} 个分子已写入 {output_mae_path}) except Exception as e: print(f 处理文件 {sdf_path} 时出错: {e}, filesys.stderr) def main(input_dir, output_base_dir): 主函数。 :param input_dir: 包含输入SDF文件的目录 :param output_base_dir: 输出文件的基础目录 # 创建输出目录 mae_dir os.path.join(output_base_dir, ‘mae_files‘) os.makedirs(mae_dir, exist_okTrue) # 查找所有SDF文件 sdf_pattern os.path.join(input_dir, ‘*.sdf‘) sdf_files glob.glob(sdf_pattern) if not sdf_files: print(f在目录 ‘{input_dir}‘ 中未找到任何.sdf文件。) return print(f找到 {len(sdf_files)} 个SDF文件。开始处理...) # 用于存储所有CSV数据的列表 all_molecule_data [] # 逐个处理文件 for sdf_file in sdf_files: process_sdf_file(sdf_file, mae_dir, all_molecule_data) # 将数据写入CSV if all_molecule_data: csv_path os.path.join(output_base_dir, ‘molecule_report.csv‘) df pd.DataFrame(all_molecule_data) # 对列进行排序让报告更易读 df df[[‘Source_SDF‘, ‘Molecule_Title‘, ‘Molecular_Weight‘, ‘Heavy_Atom_Count‘, ‘Rotatable_Bond_Count‘, ‘Output_MAE‘]] df.to_csv(csv_path, indexFalse) print(f\n分子报告已保存至: {csv_path}) print(f共处理 {len(all_molecule_data)} 个分子。) else: print(\n未成功处理任何分子数据。) if __name__ __main__: # 简单的命令行参数处理 if len(sys.argv) ! 3: print(用法: python batch_sdf_to_mae.py 输入SDF目录 输出基础目录) print(示例: python batch_sdf_to_mae.py ./raw_sdfs ./processed) sys.exit(1) input_directory sys.argv[1] output_directory sys.argv[2] # 检查输入目录是否存在 if not os.path.isdir(input_directory): print(f错误输入目录 ‘{input_directory}‘ 不存在。) sys.exit(1) main(input_directory, output_directory)5.3 脚本解析与关键点这个脚本虽然不长但涵盖了PySchrödinger实战中的多个核心模式模块化函数设计将不同的功能计算分子量、处理单个文件封装成函数使主逻辑清晰也便于复用和测试。健壮的文件遍历使用glob模块和os.path处理路径兼容不同操作系统。上下文管理器with语句用于安全地打开和关闭文件阅读器StructureReader和写入器StructureWriter确保资源被正确释放即使处理过程中发生异常。批量处理中的状态管理注意mol_count_in_file变量的使用。一个SDF文件可能包含多个分子“多记录SDF”。我们需要为同一个文件中的不同分子生成不同的标识避免标题冲突。与Pandas的集成我们将提取的数据存储为字典列表最后用Pandas的DataFrame轻松转换为CSV。这展示了PySchrödinger如何与Python科学生态无缝结合。错误处理使用try...except捕获单个文件处理过程中的异常并打印错误信息而不是让整个脚本崩溃。这在实际处理成百上千个文件时非常关键。如何运行与扩展将脚本保存并准备一个包含若干.sdf文件的测试目录./test_sdfs。在终端运行python batch_sdf_to_mae.py ./test_sdfs ./output查看./output目录你会看到mae_files文件夹内含转换后的.mae文件和molecule_report.csv。扩展思路并行处理如果文件非常多可以使用Python的multiprocessing或concurrent.futures模块并行处理多个文件大幅提升速度。更多属性你可以轻松扩展calculate_molecular_weight这样的函数添加更多的分子描述符计算比如脂水分配系数LogP的估算、氢键供体/受体数量等。Schrödinger的schrodinger.structutils.analyze模块里有很多现成的函数。条件过滤在写入MAE文件前可以加入判断。例如只保留分子量在200到500之间、且可旋转键小于10的“类药”分子。通过这个案例你已经完成了一个从数据输入、处理到输出的完整自动化流程。这不仅仅是格式转换更是一个可定制、可扩展的数据预处理管道的基础。在接下来的教程中我们将以此为基础接入真正的计算引擎比如让这些准备好的配体去和蛋白进行自动对接。
Schrödinger Python API入门：从脚本自动化到计算化学工作流构建
发布时间：2026/7/31 9:19:59
1. 从脚本小子到计算化学家为什么你需要掌握Schrödinger Python API如果你和我一样在计算化学或者药物设计的领域里摸爬滚打了一段时间大概率会经历这样一个阶段每天在Maestro的图形界面里点来点去重复着“导入结构-设置参数-提交任务-等待结果-导出数据”的循环。刚开始觉得图形界面真方便所见即所得。但项目做多了特别是需要处理几十上百个分子、进行高通量筛选或者复杂工作流自动化时这种手动操作就变成了噩梦。效率低下不说还容易出错一个手滑点错参数可能一晚上的计算就白跑了。这时候Schrödinger Python API通常我们简称为PySchrödinger就成了从“操作工”进阶为“工程师”的关键钥匙。它不是一个独立的软件而是Schrödinger Suite一个集成了分子模拟、药物设计等强大工具的商业软件套件提供给我们的一个编程接口。简单来说它允许你用Python代码去“指挥”Maestro、Glide、Prime、Desmond这些强大的计算引擎把繁琐的图形界面操作变成一行行可重复、可追溯、可批处理的脚本。我最初接触它是因为一个激进的虚拟筛选项目需要对接上万个化合物到同一个蛋白的多个构象。手动不可能。正是PySchrödinger让我在一周内完成了原本需要数月手动操作的任务并且所有参数、所有步骤都记录在脚本里可随时复现和审查。这不仅仅是效率的提升更是工作方式的革命——从依赖直觉和手工转向基于代码的、可工程化的研究流程。所以无论你是计算化学的研究生还是药物发现一线的科学家如果你已经厌倦了重复劳动渴望将创造力从机械操作中解放出来投入到更核心的科学问题中去那么这个系列就是为你准备的。我们将从“为什么需要它”开始一步步拆解这个强大工具的核心概念、基础操作和实战技巧最终让你能亲手打造属于自己的自动化研究管线。2. PySchrödinger生态全景它到底是什么能做什么在深入代码之前我们必须先搞清楚PySchrödinger在整个Schrödinger生态中的位置以及它的能力边界。这有助于我们建立正确的心理预期知道该用它解决什么问题以及哪些问题它可能不擅长。2.1 核心定位连接Python与计算引擎的桥梁首先要破除一个常见的误解PySchrödinger并不是一个用来“重新发明轮子”的独立计算库。它不提供从头实现分子力学力场、量子化学计算或者分子对接算法的新函数。相反它的核心价值在于“封装”和“驱动”。你可以把它想象成一个超级遥控器。Schrödinger Suite里的各个模块Glide对接、Prime优化、Desmond动力学等就像是一台台功能强大但操作复杂的专业设备比如高级相机、录音棚调音台。PySchrödinger就是这个统一的遥控器它定义了标准的“按键”函数和方法让你可以用Python程序来按下这些按键从而远程、精确地控制这些设备工作并获取它们产出的结果。这种设计带来了几个根本性优势自动化将图形界面的点击流程转化为脚本实现7x24小时无人值守的批量计算。可重复性脚本即记录。任何分析流程都可以被精确复现这对于科学研究至关重要。集成性可以轻松地将Schrödinger的计算能力嵌入到你用Python构建的更大数据分析流水线中与NumPy、Pandas、Scikit-learn等科学生态系统无缝衔接。灵活性可以编写复杂的逻辑条件判断、循环、自定义函数来控制计算流程这是图形界面难以实现的。2.2 核心模块与能力地图PySchrödinger的API是模块化的大致对应着Schrödinger Suite中的主要应用程序。了解这些模块你就知道了它的“武器库”里有什么。schrodinger.structure(结构模块)这是最基础、使用最频繁的模块。它提供了Structure对象用于在内存中表示和操作分子结构原子、键、坐标、属性。你可以用它来读取/写入各种文件格式.mae,.sdf,.pdb,.mol2遍历原子、修改坐标、计算简单的几何性质距离、角度、二面角、处理子结构等。几乎所有涉及分子数据输入输出的操作都从这里开始。schrodinger.application(应用模块)这是驱动计算引擎的核心。其下又有众多子模块schrodinger.application.glide用于运行Glide分子对接。你可以用代码设置搜索框、精度、打分函数等所有在Glide面板上能看到的参数。schrodinger.application.prime用于运行Prime的蛋白质结构优化、侧链预测、能量计算等。schrodinger.application.desmond用于设置和提交Desmond分子动力学模拟。schrodinger.application.ligprep用于运行LigPrep进行配体预处理加氢、生成互变异构体、电离状态等。schrodinger.application.jaguar用于运行Jaguar量子化学计算。schrodinger.job(任务管理模块)用于提交和管理计算任务。它封装了作业提交到本地机器、局域网集群或云平台上的细节。你可以用它来启动一个Glide作业并监控其运行状态等待、运行、完成、失败。schrodinger.analysis(分析模块)包含一些用于分析计算结果的工具例如分析动力学轨迹、计算结合自由能MM/GBSA等。不过很多深入的分析仍需结合第三方库如MDTraj、MDAnalysis或自定义脚本。schrodinger.utils(工具模块)提供一些通用工具函数如文件操作、日志记录等。注意PySchrödinger的官方文档有时更新不及时且某些高级功能可能缺乏示例。最可靠的学习方式往往是结合官方文档、查看Schrödinger安装目录下的示例脚本通常在$SCHRODINGER/utilities/python/或类似路径以及在实际项目中摸索。2.3 工作模式两种主要的交互方式PySchrödinger主要支持两种工作模式适用于不同场景“库”模式 (In-Process)在你的Python脚本中直接导入schrodinger模块调用其函数。这些函数会在当前Python进程内调用Schrödinger的编译库执行计算。这种方式速度快适合轻量级、不需要复杂任务管理的操作比如读取一个结构文件、进行简单的几何变换。# 示例库模式读取文件 from schrodinger.structure import StructureReader with StructureReader(ligands.mae) as reader: for st in reader: print(st.title, st.atom_total)“作业”模式 (Out-of-Process)通过schrodinger.job模块创建作业对象并提交。计算任务会作为一个独立的进程或在远程服务器上运行。这是运行耗时计算如对接、动力学的标准方式。你的脚本提交作业后可以继续执行其他逻辑或者等待作业完成。# 示例作业模式提交Glide对接 from schrodinger.application.glide import glide from schrodinger.job import jobcontrol # 创建Glide作业对象并配置参数 glide_job glide.Glide() glide_job.setInputFile(receptor.mae) glide_job.setLigandFile(ligands.mae) glide_job.setPrecision(glide.Precision.SP) # 标准精度 # 提交作业 job_id jobcontrol.submit(glide_job) print(fJob submitted with ID: {job_id}) # 可选项等待作业完成 jobcontrol.wait_for_job(job_id)理解这两种模式的区别至关重要它决定了你脚本的架构。大部分生产环境下的自动化流程都是两种模式混合使用用“库”模式准备输入文件用“作业”模式提交重型计算再用“库”模式分析输出结果。3. 环境搭建与“Hello World”迈出第一步理论说再多不如动手跑一行代码。搭建PySchrödinger的开发环境有其特殊性因为它深度依赖于Schrödinger Suite的安装。3.1 安装前提Schrödinger Suite是基石最重要的一点你必须先拥有一个正确安装并激活许可的Schrödinger Suite。PySchrödinger是随套件一起安装的不是一个可以通过pip install schrodinger获取的独立包。通常安装完成后Schrödinger会提供一个专门配置好的Python解释器。找到它Linux/macOS通常在$SCHRODINGER/utilities/python目录下。你可以通过$SCHRODINGER/run python命令直接启动这个解释器。Windows通常在Schrödinger的安装目录下例如C:\Schrodinger2024-1\utilities\python.exe。也可以在开始菜单的Schrödinger程序组里找到名为“Schrödinger Python”的快捷方式。为什么必须用它因为这个Python解释器已经预编译链接了所有Schrödinger的底层C/C库并正确设置了环境变量如SCHRODINGER。如果你使用系统自带的Python或Anaconda环境直接import schrodinger十有八九会失败报错找不到模块或库文件。3.2 配置你的开发环境以VS Code为例虽然可以用命令行直接操作但使用一个集成开发环境IDE会极大提升效率。这里以VS Code为例展示如何配置。创建项目目录为你的一系列脚本创建一个干净的文件夹例如my_pyschrodinger_projects。在VS Code中打开该文件夹。配置Python解释器按下CtrlShiftP(Windows/Linux) 或CmdShiftP(macOS)输入“Python: Select Interpreter”。选择“Enter interpreter path”然后浏览并选中Schrödinger提供的Python解释器路径例如C:\Schrodinger2024-1\utilities\python.exe。选择后VS Code右下角会显示当前使用的解释器。创建虚拟环境可选但推荐虽然Schrödinger的Python环境是独立的但为了安装一些额外的数据分析包如pandas, matplotlib最好在其基础上创建一个虚拟环境。# 在项目目录下打开终端使用Schrödinger Python # 假设Schrödinger Python路径已加入系统PATH名为‘schrodinger_python’ schrodinger_python -m venv venv # 激活虚拟环境 # Windows: .\venv\Scripts\activate # Linux/macOS: source venv/bin/activate # 在激活的虚拟环境中pip安装的包会独立存放 pip install pandas matplotlib jupyter然后在VS Code中再次选择解释器这次选择虚拟环境下的python.exe例如my_pyschrodinger_projects\venv\Scripts\python.exe。3.3 你的第一个脚本读取并查看分子让我们写一个最简单的脚本来验证环境并建立直观感受。创建一个新文件命名为hello_pyschrodinger.py。#!/usr/bin/env python 第一个PySchr?dinger脚本读取分子文件并打印基本信息。 确保你有一个可用的.mae或.sdf文件或者使用Schr?dinger自带的示例文件。 # 1. 导入核心模块 from schrodinger.structure import StructureReader, StructureWriter import sys def main(input_file): 主函数读取分子文件并打印信息。 :param input_file: 输入的分子文件路径如.mae, .sdf格式 # 2. 使用StructureReader读取文件 # StructureReader是一个迭代器可以高效地处理包含多个分子的文件 print(f正在读取文件: {input_file}) try: # ‘with‘语句确保文件被正确关闭 with StructureReader(input_file) as reader: # 3. 遍历文件中的每一个分子结构 for i, st in enumerate(reader): print(f\n--- 分子 {i1} ---) print(f 标题: {st.title}) print(f 原子总数: {st.atom_total}) print(f 化学式: {st.formula}) # 4. 访问并打印前几个原子的信息避免输出过长 print(f 前5个原子信息:) for atom in st.atom[:5]: # st.atom是一个原子列表 # atom对象包含索引、元素符号、坐标、电荷等属性 print(f 原子索引 {atom.index}: 元素 {atom.element}, f坐标 ({atom.x:.3f}, {atom.y:.3f}, {atom.z:.3f})) # 5. 演示一个简单操作计算分子量近似 # 注意这是基于原子量的简单加和未考虑同位素 mol_weight sum(atom.atom_weight for atom in st.atom) print(f 近似分子量: {mol_weight:.2f} g/mol) except FileNotFoundError: print(f错误找不到文件 {input_file}) sys.exit(1) except Exception as e: print(f读取文件时发生未知错误: {e}) sys.exit(1) if __name__ __main__: # 检查命令行参数 if len(sys.argv) ! 2: print(用法: python hello_pyschrodinger.py 分子文件路径) print(示例: python hello_pyschrodinger.py example.mae) sys.exit(1) input_file_path sys.argv[1] main(input_file_path)如何运行准备一个分子文件。如果你没有可以使用Maestro创建一个简单的分子如丙烷并保存为.mae格式或者到Schrödinger安装目录下寻找示例文件例如在$SCHRODINGER/mmshare-vX.Y/example_files/里。在终端中确保你的Python环境已激活如果用了虚拟环境然后运行python hello_pyschrodinger.py 你的分子文件.mae如果一切正常你将看到终端打印出该分子的基本信息。这段代码揭示了什么Structure对象是核心st变量代表一个完整的分子它是所有操作的起点。原子是可迭代和可索引的st.atom是一个列表你可以通过索引st.atom[0]或循环来访问每一个原子。属性访问非常直观像st.title,atom.element,atom.x这样直接访问符合Python的简洁哲学。文件读取是流式的StructureReader即使面对包含成千上万个分子的超大文件也能高效地一次处理一个不会全部加载到内存。这个简单的脚本已经包含了PySchrödinger最基础、最常用的模式。运行成功意味着你的环境已经就绪我们可以开始探索更强大的功能了。4. 深入Structure对象分子数据的“瑞士军刀”上一节我们接触了Structure对象。如果说PySchrödinger是一套乐高那Structure就是最基础、最重要的那块积木。几乎所有工作都始于读取一个结构终于写入或分析一个结构。因此彻底理解Structure对象至关重要。4.1 结构属性不仅仅是原子坐标一个Structure对象包含的信息远不止原子列表。它由多个“层面”的数据构成原子层面 (Atom Level)这是最核心的数据。每个原子是一个_Atom对象通常我们直接叫它atom其关键属性包括index: 原子在结构中的唯一整数索引从1开始。element: 元素符号字符串如 ‘C‘ ‘N‘ ‘O‘。x, y, z: 三维空间坐标浮点数。charge: 部分电荷浮点数。atom_type: 力场原子类型字符串。pdbname: PDB格式中的原子名称如 ‘ CA ‘。resnum: 残基序号整数。chain: 链标识符字符串如 ‘A‘。inscode: 插入码字符串用于处理PDB中残基序号不连续的情况。结构层面 (Structure Level)整个分子或体系的全局属性。title: 结构的标题通常从文件名或注释行读取。property: 这是一个类似字典的接口用于访问和修改存储在结构文件中的各种属性。这是Structure对象最强大也最容易让人困惑的特性之一。分子属性如r_mmod_Potential_Energy(分子力学势能)s_m_title(标题)i_m_ct_format(格式版本)等。原子属性每个原子也可以有独立的属性通过atom.property访问如r_j_Atm_polarizability(原子极化率)。这些属性键key通常是schrodinger内部定义的字符串很多以r_(实数),i_(整数),s_(字符串),b_(布尔值) 开头表示数据类型。化学信息层面bond: 键的列表。每个键对象连接两个原子索引并包含键级单键、双键等信息。formula: 化学式字符串。4.2 属性Property系统详解与实战property系统是Schrödinger文件格式如.mae存储额外数据的核心机制。理解它你才能自由地读写计算产生的各种数据如对接打分、能量值、RMSD等。如何查看一个结构有哪些属性from schrodinger.structure import StructureReader with StructureReader(‘complex.mae‘) as reader: st next(reader) # 读取第一个结构 # 获取所有结构级别的属性名 prop_names st.property.keys() print(结构属性列表:) for name in prop_names: # 获取属性值及其数据类型 value st.property[name] print(f {name}: {value} (类型: {type(value).__name__})) # 查看第一个原子的属性如果有的话 if st.atom_total 0: atom st.atom[0] atom_prop_names atom.property.keys() print(f\n第一个原子的属性:) for name in atom_prop_names[:5]: # 只打印前5个避免过多 print(f {name}: {atom.property[name]})如何读写属性# 写入一个新的结构属性 st.property[‘s_my_custom_notes‘] ‘This is a manually added note.‘ # 写入一个新的原子属性为每个原子添加 for atom in st.atom: atom.property[‘r_my_custom_charge‘] 0.0 # 假设我们自定义一个电荷 # 读取Glide对接结果中常见的打分属性 if ‘r_i_docking_score‘ in st.property: docking_score st.property[‘r_i_docking_score‘] print(fDocking Score: {docking_score})踩坑点1属性名的“黑盒”性。很多属性名如r_i_docking_score是Schrödinger内部定义的文档可能不全。最可靠的方法是先用一个已知包含所需数据的文件比如一个Glide对接输出文件运行上面的查看代码找到确切的属性名然后在你的脚本中使用它。踩坑点2属性类型匹配。当你写入属性时键的前缀最好与值的类型匹配s_对应字符串i_对应整数r_对应浮点数b_对应布尔值虽然不匹配有时也能工作但可能在其他Schrödinger工具中引发问题。4.3 常用结构操作选择、修改与遍历掌握了数据存取下一步就是操作。原子选择与索引# 通过索引访问原子索引从1开始 atom_5 st.atom[5] # 获取第6个原子因为索引5对应第6个 # 遍历所有原子 for atom in st.atom: if atom.element ‘O‘: print(f找到氧原子索引: {atom.index}) # 使用列表推导式进行筛选 carbon_atoms [atom for atom in st.atom if atom.element ‘C‘] print(f碳原子数量: {len(carbon_atoms)})修改结构# 修改原子坐标例如将第一个原子移动到原点 if st.atom_total 0: st.atom[0].x 0.0 st.atom[0].y 0.0 st.atom[0].z 0.0 # 添加氢原子调用Schr?dinger的加氢算法 # 注意这需要导入特定模块且可能依赖力场 from schrodinger.structutils import build build.add_hydrogens(st) # 删除水分子示例删除链名为‘W‘的残基 # 注意直接删除原子会改变后续原子的索引需要小心处理或从后往前删 atoms_to_delete [] for atom in st.atom: if atom.chain ‘W‘: # 假设水分子在链W atoms_to_delete.append(atom.index) # 按索引从大到小删除避免索引错乱 for index in sorted(atoms_to_delete, reverseTrue): st.deleteAtom(index)几何计算from schrodinger.structutils import measure # 计算两个原子间的距离假设原子索引1和10 if st.atom_total 10: dist measure.measure_distance(st, 1, 10) print(f原子1和10之间的距离: {dist:.3f} ?) # 计算三个原子形成的角度原子索引1510 angle measure.measure_angle(st, 1, 5, 10) print(f角度(1-5-10): {angle:.2f} 度) # 计算四个原子形成的二面角扭转角 dihedral measure.measure_dihedral(st, 1, 5, 10, 15) print(f二面角(1-5-10-15): {dihedral:.2f} 度)写入文件from schrodinger.structure import StructureWriter # 写入单个结构到新文件 with StructureWriter(‘modified_structure.mae‘) as writer: writer.append(st) # 将当前结构对象写入文件 # 批量写入多个结构例如筛选后的结果 good_structures [] # 假设这是一个经过筛选的Structure对象列表 with StructureWriter(‘filtered_results.mae‘) as writer: for good_st in good_structures: writer.append(good_st)通过熟练运用Structure对象及其相关工具structutils你就能完成分子数据的预处理、后处理、分析和转换等绝大部分日常工作。这是构建任何自动化流程的基石。5. 第一个实战案例批量分子文件格式转换与信息提取现在让我们把前面学到的知识组合起来解决一个实际工作中经常遇到的问题批量处理一堆分子文件将它们从一种格式转换成另一种格式并同时提取一些关键信息保存到表格中。假设你从公共数据库下载了1000个化合物的.sdf文件你需要将它们全部转换成Maestro格式.mae因为后续的Glide对接需要这个格式。从每个分子中提取一些基本信息比如分子量、重原子数、可旋转键数并保存到一个CSV文件中方便快速筛选。我们将一步步构建这个脚本。5.1 设计脚本逻辑输入一个包含多个.sdf文件的目录。处理 a. 遍历目录中的所有.sdf文件。 b. 对于每个文件使用StructureReader读取其中的所有分子。 c. 对每个分子结构st i. 计算所需属性分子量、重原子数、可旋转键数。 ii. 将结构写入新的.mae文件可以每个分子一个文件或合并成一个文件。 iii. 将提取的属性信息存储到一个列表里。输出 a. 转换后的.mae文件。 b. 一个CSV文件包含文件名、分子标题、以及计算出的各项属性。5.2 代码实现创建一个新文件命名为batch_sdf_to_mae.py。#!/usr/bin/env python 批量SDF转MAE并提取分子信息脚本。 功能将指定文件夹内所有.sdf文件转换为.mae文件并生成包含分子信息的CSV报告。 import os import sys import glob from schrodinger.structure import StructureReader, StructureWriter from schrodinger.structutils import analyze import pandas as pd def calculate_molecular_weight(st): 计算近似分子量基于原子量加和。 :param st: Structure对象 :return: 分子量 (float) return sum(atom.atom_weight for atom in st.atom) def count_heavy_atoms(st): 计算重原子非氢原子数量。 :param st: Structure对象 :return: 重原子数 (int) return sum(1 for atom in st.atom if atom.element ! ‘H‘) def count_rotatable_bonds(st): 计算可旋转键的数量粗略估计忽略环内键和末端键。 使用Schr?dinger内置的analyze模块。 :param st: Structure对象 :return: 可旋转键数 (int) # evaluate_rotatable_bonds返回一个元组第一个元素是数量 return analyze.evaluate_rotatable_bonds(st)[0] def process_sdf_file(sdf_path, output_mae_dir, output_csv_data): 处理单个SDF文件。 :param sdf_path: 输入的SDF文件路径 :param output_mae_dir: 输出MAE文件的目录 :param output_csv_data: 用于存储CSV数据的列表 base_name os.path.splitext(os.path.basename(sdf_path))[0] output_mae_path os.path.join(output_mae_dir, f{base_name}.mae) print(f处理文件: {sdf_path}) try: with StructureReader(sdf_path) as reader, \ StructureWriter(output_mae_path) as writer: mol_count_in_file 0 for st in reader: mol_count_in_file 1 # 为每个分子生成一个唯一的标识如果文件中有多个分子 if mol_count_in_file 1: mol_title_for_csv f{base_name}_mol{mol_count_in_file} # 在结构标题中也加上标识避免重复 st.title f{st.title}_{mol_count_in_file} if st.title else mol_title_for_csv else: mol_title_for_csv base_name # 1. 计算分子属性 mol_weight calculate_molecular_weight(st) heavy_atoms count_heavy_atoms(st) rot_bonds count_rotatable_bonds(st) # 2. 将信息添加到CSV数据列表 output_csv_data.append({ ‘Source_SDF‘: os.path.basename(sdf_path), ‘Molecule_Title‘: st.title, ‘Molecular_Weight‘: round(mol_weight, 2), ‘Heavy_Atom_Count‘: heavy_atoms, ‘Rotatable_Bond_Count‘: rot_bonds, ‘Output_MAE‘: os.path.basename(output_mae_path) }) # 3. 将结构写入MAE文件 # 注意这里将所有分子写入同一个MAE文件。如果希望每个分子单独文件需调整逻辑。 writer.append(st) print(f 完成。发现 {mol_count_in_file} 个分子已写入 {output_mae_path}) except Exception as e: print(f 处理文件 {sdf_path} 时出错: {e}, filesys.stderr) def main(input_dir, output_base_dir): 主函数。 :param input_dir: 包含输入SDF文件的目录 :param output_base_dir: 输出文件的基础目录 # 创建输出目录 mae_dir os.path.join(output_base_dir, ‘mae_files‘) os.makedirs(mae_dir, exist_okTrue) # 查找所有SDF文件 sdf_pattern os.path.join(input_dir, ‘*.sdf‘) sdf_files glob.glob(sdf_pattern) if not sdf_files: print(f在目录 ‘{input_dir}‘ 中未找到任何.sdf文件。) return print(f找到 {len(sdf_files)} 个SDF文件。开始处理...) # 用于存储所有CSV数据的列表 all_molecule_data [] # 逐个处理文件 for sdf_file in sdf_files: process_sdf_file(sdf_file, mae_dir, all_molecule_data) # 将数据写入CSV if all_molecule_data: csv_path os.path.join(output_base_dir, ‘molecule_report.csv‘) df pd.DataFrame(all_molecule_data) # 对列进行排序让报告更易读 df df[[‘Source_SDF‘, ‘Molecule_Title‘, ‘Molecular_Weight‘, ‘Heavy_Atom_Count‘, ‘Rotatable_Bond_Count‘, ‘Output_MAE‘]] df.to_csv(csv_path, indexFalse) print(f\n分子报告已保存至: {csv_path}) print(f共处理 {len(all_molecule_data)} 个分子。) else: print(\n未成功处理任何分子数据。) if __name__ __main__: # 简单的命令行参数处理 if len(sys.argv) ! 3: print(用法: python batch_sdf_to_mae.py 输入SDF目录 输出基础目录) print(示例: python batch_sdf_to_mae.py ./raw_sdfs ./processed) sys.exit(1) input_directory sys.argv[1] output_directory sys.argv[2] # 检查输入目录是否存在 if not os.path.isdir(input_directory): print(f错误输入目录 ‘{input_directory}‘ 不存在。) sys.exit(1) main(input_directory, output_directory)5.3 脚本解析与关键点这个脚本虽然不长但涵盖了PySchrödinger实战中的多个核心模式模块化函数设计将不同的功能计算分子量、处理单个文件封装成函数使主逻辑清晰也便于复用和测试。健壮的文件遍历使用glob模块和os.path处理路径兼容不同操作系统。上下文管理器with语句用于安全地打开和关闭文件阅读器StructureReader和写入器StructureWriter确保资源被正确释放即使处理过程中发生异常。批量处理中的状态管理注意mol_count_in_file变量的使用。一个SDF文件可能包含多个分子“多记录SDF”。我们需要为同一个文件中的不同分子生成不同的标识避免标题冲突。与Pandas的集成我们将提取的数据存储为字典列表最后用Pandas的DataFrame轻松转换为CSV。这展示了PySchrödinger如何与Python科学生态无缝结合。错误处理使用try...except捕获单个文件处理过程中的异常并打印错误信息而不是让整个脚本崩溃。这在实际处理成百上千个文件时非常关键。如何运行与扩展将脚本保存并准备一个包含若干.sdf文件的测试目录./test_sdfs。在终端运行python batch_sdf_to_mae.py ./test_sdfs ./output查看./output目录你会看到mae_files文件夹内含转换后的.mae文件和molecule_report.csv。扩展思路并行处理如果文件非常多可以使用Python的multiprocessing或concurrent.futures模块并行处理多个文件大幅提升速度。更多属性你可以轻松扩展calculate_molecular_weight这样的函数添加更多的分子描述符计算比如脂水分配系数LogP的估算、氢键供体/受体数量等。Schrödinger的schrodinger.structutils.analyze模块里有很多现成的函数。条件过滤在写入MAE文件前可以加入判断。例如只保留分子量在200到500之间、且可旋转键小于10的“类药”分子。通过这个案例你已经完成了一个从数据输入、处理到输出的完整自动化流程。这不仅仅是格式转换更是一个可定制、可扩展的数据预处理管道的基础。在接下来的教程中我们将以此为基础接入真正的计算引擎比如让这些准备好的配体去和蛋白进行自动对接。
