获取整数n并使用字符串格式打印pi到n位的函数

import math

def print_pi(n):
    if n <= 0:
        return "请输入大于0的整数"
    
    # 计算pi的值
    pi = str(math.pi)
    
    # 检查n是否超过pi的位数，如果超过则打印全部位数
    if n > len(pi):
        return pi
    
    # 截取pi到n位并打印
    return pi[:n+2]  # 加2是为了包括小数点和整数部分的数字

# 示例调用
n = int(input("请输入要打印的位数："))
result = print_pi(n)
print(result)