def split_list(a_list, n):
    """
    split one list into n lists of equal size. In this case, we are splitting
    the list that represents the policy of a single each control structure
    so that each time step is separate
    """
    portions = len(a_list)/n
    split_lists = []
    for i in range(n):
        split_lists.append(a_list[i*portions: (i+1)*portions])
    return split_lists