w = np.vectorize(lambda i: self.mol.GetAtomWithIdx(int(i)).GetMass())(ids)