如何打印根据变量改变大小的菱形形状？

def print_diamond(size):
    if size % 2 == 0:
        size += 1  # 确保菱形的大小为奇数

    # 打印上半部分菱形
    for i in range(1, size // 2 + 2):
        print(" " * (size // 2 + 1 - i) + "*" * (2 * i - 1))

    # 打印下半部分菱形
    for i in range(size // 2, 0, -1):
        print(" " * (size // 2 + 1 - i) + "*" * (2 * i - 1))

# 测试
size = int(input("请输入菱形的大小："))
print_diamond(size)