python_test_29

43.请定义两个函数，一个函数画正方形，一个函数画三角形，并且可以从键盘输入值来决定画正方形还是画三角形以及决定是否退出程序。

def square(n):
    for i in range(n):
        print("* " * n)


def triangle(n):
    for i in range(n):
        print("* " * (i + 1))


if __name__ == '__main__':
    while True:
        select = input("请输入要画的形状（正方形或者直角三角形）：")
        if select == "正方形":
            square(10)
        elif select == "直角三角形":
            triangle(10)
        else:
            break