Untitled

def rysuj_trapez_rownoramienny(n):
    k = 1 + 2 * (n - 1)
    for i in range(n):
        print(" " * (n - i - 1) + "*" * k)
        k += 2

def rysuj_trapez_nierownoramienny(n, k):
    for i in range(n):
        print(" " * i + "*" * k)
        k += 2

wybor = input("Wybierz rodzaj trapezu (równoramienny - 'r', nierównoramienny - 'n'): ")

if wybor == 'r':
    wysokosc = int(input("Podaj wysokość trapezu: "))
    rysuj_trapez_rownoramienny(wysokosc)
elif wybor == 'n':
    wysokosc = int(input("Podaj wysokość trapezu: "))
    dlugosc_gornej_podstawy = int(input("Podaj długość górnej podstawy trapezu: "))
    rysuj_trapez_nierownoramienny(wysokosc, dlugosc_gornej_podstawy)
else:
    print("Niepoprawny wybór.")