在Jupyter-lab中使用RDKit画分子2D图

在做完分子对接后，想看看筛选后的分子的结构。因此想利用Jupyter-lab来画分子的2D图。

1. 安装Jupyter-lab与RDKit

• 系统：Win11
• 已安装conda

RDKit 是一个功能强大、灵活易用的化学信息学工具包，广泛应用于药物发现、化学生物学、材料科学等领域。RDKit 提供了丰富的化学信息可视化功能，用户可以通过RDKit 生成分子结构图、化学反应图、药物分子的三维结构等，方便进行结果展示和分析。

1.1 安装RDKit

#创建并安装RDKit的conda环境
conda create -c conda-forge -n my-rdkit-env rdkit

#激活RDKit的虚拟环境
conda activate my-rdkit-env

不知道为啥，这一步很慢，不知道是不是conda channel的问题。但总归是等了一段时间就成功了。参考自官方教程。

1.2 安装Jupyter-lab

Jupyter Lab是Jupyter Notebook的升级版本，可以一个窗口中同时打开多个 Notebook、代码编辑器、终端、文件浏览器等，更方便地进行多任务处理和工作流程管理。Jupyter Lab集成了丰富的文档编辑器，包括 Markdown、LaTeX 等，用户可以方便地编写和编辑文档，同时可以通过插件支持更多的文档格式和扩展功能。最主要的是Jupyter Lab提供了丰富的可视化功能，用户可以通过插件支持各种图表库和数据可视化工具，实现更丰富和复杂的数据分析和可视化。

使用pip安装

pip install jupyterlab

参考自官方安装教程。

2. 小分子（SDF格式）作图

2.1 画单个SDF格式的分子图

以他莫昔芬（Tamoxifen）小分子为例，首先从PubChem网站下载该分子的三维结构的SDF结构文件。

如法炮制，再下载其余三个小分子：阿司匹林（Aspirin），莫洛昔康（Molnupiravir），瑞德西韦（Remdesivir）。将四个分子保存在sdf/文件夹中。

在CMD中跳转至cd sdf/文件夹路径下，打开Jupyter-lab。

from rdkit import Chem
from rdkit.Chem import AllChem
from rdkit.Chem import Draw

#读取sdf文件
m1 = Chem.MolFromMolFile('sdf/Conformer3D_COMPOUND_CID_121304016.sdf')

#通过调用AllChem.Compute2DCoords(m1)计算出m1分子相应的二维坐标，并将其更新到分子对象中
AllChem.Compute2DCoords(m1)

#判断m1分子是否读取成功
m1 is None

#Draw.MolToImage()函数画出分子的2D图像
Draw.MolToImage(m1)

运行后结果如下：

• 读取单个的SDF文件可以用Chem.MolFromMolFile()，尽管这个主要用于读取MOL格式的分子文件。
• 读取一个存有多个分子集的SDF文件，主要用Chem.SDMolSupplier()。
• Chem.MolFromMolFile()无法读取PDB格式的分子文件，因此小分子文件读取主推SDF格式。

2.2 画多个SDF格式的分子图

读取sdf/文件夹下的多个SDF分子，且将分子的2D图放在一张图中。

import os
import glob

# 指定小分子的存储路径
path = r'sdf/'

# 使用os.path.join构建路径，获取所有.sdf文件的路径列表
sdf_files = glob.glob(os.path.join(path, '*.sdf'))

# 将所有.sdf文件保存在一个列表ms中
ms = []
for sdf_file in sdf_files:
    m = Chem.MolFromMolFile(sdf_file)
    ms.append(m)

#计算出列表中分子的二维坐标，并将其更新到分子对象中  
for m in ms: tmp=AllChem.Compute2DCoords(m)

# MolsToGridImage()函数可以将4个分子画在一张图上
# molsPerRow=4参数设置4个小分子并排成一列
# subImgSize=(500,500)设置每个小分子的图像尺寸为500x500大小
# legends=x.GetProp("_Name")获取小分子的名称作为图例
img=Draw.MolsToGridImage(ms[:4],molsPerRow=4,subImgSize=(500,500),legends=[x.GetProp("_Name") for x in ms[:4]], returnPNG=False) 

# 保存分子图像为PNG格式文件在sdf/文件夹下
img.save('./Mol_4.png')

img

结果如下所示：

在Draw.MolsToGridImage函数中设置returnPNG=False参数，主要用于在保存画出的图片时img.save(）的报错：

---

AttributeError Traceback (most recent call last) Cell In[23], line 20
17 for m in ms: tmp=AllChem.Compute2DCoords(m)
19 img=Draw.MolsToGridImage(ms[:4],molsPerRow=4,subImgSize=(500,500),legends=[x.GetProp(“_Name”)
for x in ms[:4]])
—> 20 img.save(‘./Mol12_20.png’)
21 img

AttributeError: ‘Image’ object has no attribute ‘save’