Nested Triangle In Python

My assingment At each level the complete triangle for the previous level is placed into an extra outer triangle. The user should be asked to input the two characters to be used an

Solution 1:

The dictionary can be created using a simple function:

def create_tri_dict(tri_chars, tri_height):
    level_str = {0:tri_chars[0]}

    for i in range(1,tri_height):
        level_length = i *2 +1
        tri_char = tri_chars[i%2]
        level_str[i] = level_str[i-1] + '\n' + tri_char * level_length
    return level_str

Then the main logic of your program could be:

tri_chars = input('Input triangle characters: ')
tri_length = int(input('Input triangle base length: '))
tri_height = (tri_length + 1)//2if tri_length %2 == 0:
    raise Exception('Triangle base length not odd')
tri_dict = create_tri_dict(tri_chars, tri_length)

Then to print the final 3(?) triangles:

print(tri_dict[tri_height-2])
print(tri_dict[tri_height-1])
print(tri_dict[tri_height])