Skip to content

Instantly share code, notes, and snippets.

@aced125
aced125 / create_fingerprint.py
Last active December 20, 2019 00:39
RDKit fingerprint implementation
from rdkit import Chem
from rdkit.Chem.AllChem import GetMorganFingerprintAsBitVect
from rdkit.DataStructs.cDataStructs import ConvertToNumpyArray
import numpy as np
atorvastatin_smiles = 'O=C(O)C[C@H](O)C[C@H](O)CCn2c(c(c(c2c1ccc(F)cc1)c3ccccc3)C(=O)Nc4ccccc4)C(C)C'
atorvastatin = Chem.MolFromSmiles(atorvastatin_smiles) # Atorvastatin (aka Lipitor) is one of the world's best-selling drugs.
fingerprint = GetMorganFingerprintAsBitVect(atorvastatin, radius=2, nBits=2048)