Get position of closing bracket if the poisition of the opening bracket is given.

get_closing_bracket.py

def get_closing_index(string, index):
    """
    Get position of closing bracket of a opening
    bracket given by `index`.
    
    Parameters
    ----------
    
    string: str;
        Input text.
        
    index: int;
        Position of opening bracket
        
    Returns
    -------
    
    closing_index: int;
        Poisition of closing bracket    
    """
    # Check if position given corrsponds to a open bracket
    assert string[index] == '[', 'index does not point to opening bracket'

    
    control = 0 # Variable to control number of brackets
    
    # Note that we start the loop on index given forward
    for n, char in enumerate(string[index:]):
        if char == '[':
            control += 1
        elif char == ']':
            control -= 1

        # If stack is empty, returns position
        if control == 0:
            return index+n