PTA Python 7-1 求圆周率pi的估算值

输入一个n 表示项数，使用以下公式求圆周率π 的估算值：
π=12​(1−3×31​+5×321​−7×331​+...)

输入格式:

输入一个n

输出格式:

输出π 的估算值

输入样例1:

在这里给出一组输入。例如：

1

输出样例1:

在这里给出相应的输出。例如：

pi = 3.0792014356780038

输入样例2:

在这里给出一组输入。例如：

2

输出样例2:

在这里给出相应的输出。例如：

pi = 3.156181471569954

答案:

import math

n = eval(input())
sum = 0


def add(n1):
    if n1 % 2 == 1:
        return -(1 / ((2 * n1 + 1) * 3 ** n1))
    else:
        return 1 / ((2 * n1 + 1) * 3 ** n1)


for x in range(1, n + 1):
    sum += add(x)
sum = math.sqrt(12) * (1 + sum)
print(f"pi = {sum}")