Also, please can someone adjust or give me advice on how to look at the second order of parenthesis. Same process as this, but with only parenthesis in second order (this code is first order). Can you make it so I can easily adjust? By second order I mean (()). for example C(C(C))C. Everything except The C with 2 brackets around is 0. Also, the same conditions occur. Much appreciated.
Hope you are well. I have this code where I am trying to index within a parenthesis. I want all atoms that are not in parenthesis (branched) to be 0. For example
CCC(C)CC
[0 0 0 1 0 0]
CC(CCC)CC
[0 0 1 2 3 0 0]
CC(C)(C)C
[0 0 1 1 0]
CC(CCC)(C)C
[0 0 1 2 3 1 0]
As you can see from the above examples, I am counting the number of atoms within the parenthesis, however any nested parenthesis (a branched atom within a branch) is given the value of the atom before the branch (the atom without a parenthesis around it).
Such as C(C(C)C)C would have [0 1 1 2 0].
This code works for all cases except ones such as these. Below are my desired outputs, incorrect output and my code. Thanks
Desired output
CCC(CC(C)CC)(C)C [0, 0, 0, 1, 2, 2, 3, 4, 1, 0]
^ ^
Incorrect output
CCC(CC(C)CC)(C)C [0, 0, 0, 1, 2, 2, 3, 4, 4, 0]
^ ^
import pandas as pd
from rdkit import Chem
def smile_grouping(s):
group_counter = 1
res = []
open_brackets = 0
branch_start_index = None
last_non_nested_group = None
for i, letter in enumerate(s):
if letter == '(':
open_brackets += 1
if open_brackets == 1:
branch_start_index = i
if last_non_nested_group is not None:
group_counter = last_non_nested_group + 1
elif letter == ')':
if open_brackets == 1:
last_non_nested_group = None
open_brackets -= 1
if open_brackets == 0:
branch_start_index = None
elif letter not in ['[', ']', '+', '-', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'i', 'e']:
if open_brackets == 1:
if branch_start_index is not None and branch_start_index + 1 != i:
group_counter += 1
res.append(group_counter)
last_non_nested_group = group_counter
elif open_brackets == 0:
res.append(0)
elif open_brackets > 1:
res.append(last_non_nested_group)
mol = Chem.MolFromSmiles(s)
num_atoms = mol.GetNumAtoms()
while len(res) < num_atoms:
res.append(0)
return res
df = pd.DataFrame(
{'SMILES': [
"CCC(CC(C)CC)(C)C",
"CCC[I+](C)(C)C",
"CCC(CCC(C))C"
]})
df['Indexed SMILES'] = df['SMILES'].apply(smile_grouping)
print(df)
You need a way to reset group counter when it leaves the last close bracket. I added one to your open brackets logic as such:
if open_brackets == 1:
branch_start_index = i
if last_non_nested_group is not None:
group_counter = last_non_nested_group + 1
else:
group_counter = 1
This resets the group counter after it completes the final close bracket and enters a new open bracket.