Excel formula: Extracts any element n in a string separated by commas

excel_cell_extractor.py

import pandas as pd

def generate_excel_formula(n):
  """
  generate_excel_formula(n) --> Generates any excel formula that find an element in a substring:
  | data                | 1    | 2      | 3       | 4 |
  |---------------------|------|--------|---------|---|
  | dogs, cats, monkeys | dogs | cats   | monkeys |   |
  | dogs, whales        | dogs | whales |         |   |
  |                     |      |        |         |   |
  """
    if n == 1:
        return '=TRIM(LEFT(A2,FIND(",",A2&",")-1))'
    else:
        return f'=IFERROR(TRIM(MID(A2,FIND("#",SUBSTITUTE(A2,",","#",{n-1}))+1,FIND("#",SUBSTITUTE(A2,",","#",{n}))-FIND("#",SUBSTITUTE(A2,",","#",{n-1}))-1)),"")'

# Ejemplo de uso

for i in range(1,7):
    
    print(f"{generate_excel_formula(i)}")