Python 输出图形

根据输入的数字打印出菱形图案，如下图分别为输入5和9的对应图案

def fun(n):
    for i in range(1,n):
        print(' '*(n-i),'* '*i)
    print('',"* "*n)
    for i in range(1,n):
        print(' '*i,'* '*(n-i))


def main():
    a = int(input('请输入菱形边长：'))
    fun(a)

if __name__ == '__main__':
    main()

a=int(input('输入行数'))
for i in range(0,a):
    for y in range(0,a-i):
        print(' ',end="")
    s = '* ' * i
    print(s)
for i in range(0,a):
    for x in range(0,i):
        print(' ', end="")
    t = '* ' * (a-i)
    print(t)