Выделение функций улыбки с помощью rdkit ⇐ Python

[Anonymous]

Код: Выделить всё

from rdkit import Chem
from rdkit.Chem import ChemicalFeatures
from rdkit import rdBase
from rdkit.RDPaths import RDDataDir
# from rdkit.Chem.Draw import IPythonConsole
from rdkit.Chem import Draw
from rdkit.Chem import AllChem
import os

print(rdBase.rdkitVersion)

fdefFile = os.path.join(RDDataDir,'BaseFeatures.fdef')
featFact = ChemicalFeatures.BuildFeatureFactory(fdefFile)

mols = [Chem.MolFromSmiles('C1=NC(=C2C(=N1)N(C=N2)C3C(C(C(O3)CO)O)O)N')]

featslists = [featFact.GetFeaturesForMol(mol) for mol in mols]

def drawp4core(mol, feats):
atoms_list = {}
for feat in feats:
atom_ids = feat.GetAtomIds()
feat_type = feat.GetType()
atoms_list[feat_type] = atom_ids
return Draw.MolsToGridImage([mol]*len(atoms_list), legends=list(atoms_list.keys()), highlightAtomLists=list(atoms_list.values()))

im = drawp4core(mols[0], featslists[0])

print(type(im))

im.show()

Я пытаюсь выделить фармакофорные свойства малых молекул с помощью RDKIT. У меня есть следующий код (скопирован из ссылки, проверьте его на наличие полного воспроизводимого кода).
Это работает, но все основные моменты находятся на отдельных изображениях, как в примере. Я хочу, чтобы все они были наложены друг на друга, надеюсь, в разных цветах.



Подробнее здесь: https://stackoverflow.com/questions/757 ... with-rdkit