如何避免或修复Pybel对CSO3基团的错误加氢问题

导入经Schrodinger Maestro编辑的含CSO3基团的PDB文件到Pybel后，调用addh()加氢时，硫原子和其中一个氧原子被错误加上氢；但用Maestro加隐式氢完全正常。因为PDB不存储共振信息，我已经手动给两个硫氧键设置了双键，但尝试删除多余氢的代码直接触发段错误（exit code 139，SIGSEGV），需要解决加氢错误或者实现安全删氢的方案。

思路1：先修正键级再加氢，从根源避免错误 #

Pybel的addh()默认依赖键级判断加氢位点，CSO3的硫是+4价（两个双键氧、一个单键氧、一个碳），所以先手动给CSO3的S-O键设置正确键级，再加氢就能避免错误。

示例代码：

import pybel

# 读取PDB文件
mol = next(pybel.readfile("pdb", "your_cso3.pdb"))

# 定位CSO3残基并修正键级
for residue in mol.OBMol.GetResidues():
    if residue.GetName() == "CSO3":
        s_atom = None
        o_atoms = []
        # 先找出残基里的S和所有O原子
        for atom in residue.GetAtoms():
            sym = atom.GetSymbol()
            if sym == "S":
                s_atom = atom
            elif sym == "O":
                o_atoms.append(atom)
        # 给前两个O设双键，第三个设单键（对应CSO3的共振结构）
        for idx, o_atom in enumerate(o_atoms):
            bond = mol.OBMol.GetBond(s_atom, o_atom)
            if bond:
                bond.SetBondOrder(2 if idx < 2 else 1)

# 现在再加氢，用OpenBabel的原生方法更可控
mol.OBMol.AddHydrogens(False, False, 7.4)  # 参数：是否加非极性氢，是否加极性氢，pH值
# 或者用Pybel的addh()，此时键级正确不会加错
# mol.addh()

# 保存修正后的文件
mol.write("pdb", "fixed_cso3.pdb", overwrite=True)

思路2：安全删除多余氢，避免段错误 #

段错误大多是因为遍历原子时直接删除，导致迭代器失效。正确做法是先收集所有要删除的氢原子，再批量删除。

示例代码：

import pybel

mol = next(pybel.readfile("pdb", "your_cso3.pdb"))
mol.addh()  # 先执行加氢，得到带错误氢的分子

# 收集要删除的氢：连接在S上的氢，或CSO3残基里O上的氢
to_remove = []
for atom in mol:
    if atom.atomicnum == 1:  # 筛选氢原子
        neighbor = list(atom.neighbors)[0]  # 氢只有一个邻居
        # 判断邻居是S，或者是CSO3残基里的O
        if neighbor.atomicnum == 16 or (neighbor.atomicnum == 8 and neighbor.parent.name == "CSO3"):
            to_remove.append(atom.OBAtom)  # 存OBAtom对象，方便后续删除

# 批量删除氢原子
for h_atom in to_remove:
    mol.OBMol.DeleteAtom(h_atom)

# 保存结果
mol.write("pdb", "fixed_cso3.pdb", overwrite=True)

额外提示 #

• 优先用思路1，从根源避免加氢错误，比事后删氢更稳妥
• 如果CSO3残基名称不是"CSO3"，自行调整代码里的残基名称匹配
• OpenBabel的AddHydrogens方法可以通过参数控制加氢逻辑，比如指定pH值适配不同环境下的质子化状态

内容的提问来源于stack exchange，提问作者Rubber Duck