python实现用*输出一个正六边形

用 * 输出一个正六边形，输入一个整数 n 代表输出的正六边形的边的长度 (*的数目)

代码::

n = int(input("input n:"))
for i in range(n, 2 * n):
    if i <= 2 * n - 1:
        print((2 * n - 1 - i) * ' ', end='')
    print(i * ' *')
a = 1
for i in range(2 * n - 2, n - 1, -1):
    while i >= 2:
        print(a * ' ', end='')
        a += 1
        break
    print(i * ' *')

n = 3

n = 4

n = 5