Clemens Isert cisert
cisert
/ highlight_atoms_3d.py
Created
February 9, 2022 10:48
Draw 3D molecule with highlighted atoms
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import py3Dmol | |
| from rdkit import Chem | |
| def draw_mol_with_highlights(mol, hit_ats, style=None): | |
| """Draw molecule in 3D with highlighted atoms. | |
| Parameters | |
| ---------- | |
| mol : RDKit molecule | |
| hit_ats : tuple of tuples |