from rdkit import Chem from rdkit.Chem import Draw, rdDepictor from rdkit.Chem.Draw import SimilarityMaps from PIL import Image import numpy as np mol = Chem.MolFromSmiles('COc1cccc2cc(C(=O)NCCCCN3CCN(c4cccc5nccnc54)CC3)oc21') refmol = Chem.MolFromSmiles('CCCN(CCCCN1CCN(c2ccccc2OC)CC1)Cc1ccc2ccccc2c1') fp = SimilarityMaps.GetMorganFingerprint(mol, fpType='bv')#生成Morgan指纹 rdDepictor.Compute2DCoords(mol)# # 获取分子对象的2D坐标 fig, maxweight = SimilarityMaps.GetSimilarityMapForFingerprint(mol, refmol, SimilarityMaps.GetMorganFingerprint)# 绘制指纹到分子对象上 fig.canvas.draw() # 将图片缩放为高清 X = np.array(fig.canvas.renderer.buffer_rgba()) im = Image.fromarray(X) im = im.resize((800, 800),resample=Image.LANCZOS) im.save('E:/masdh/map.svg')# 更改这段代码为可以保存svg图片保存
时间: 2024-02-25 20:59:24 浏览: 415
chemkin-import_chem.inp
你可以尝试将最后一行的代码修改为以下内容:
```python
im.save('E:/masdh/map.svg', format='svg')
```
这样可以显式地指定保存的文件格式为SVG,避免出现文件扩展名不被识别的问题。
阅读全文