课后习题，根据提示写出来的程序，对python的语法还不是是熟悉，对自己写的代码不满意，争取后期修改，或者大神们提出更好的代码
# -*- coding: utf-8 -*-
import itertools
'''
计算圆周率可以根据公式：
利用Python提供的itertools模块，我们来计算这个序列的前N项和：
'''
def pi(N):
' 计算pi的值 '
# step 1: 创建一个奇数序列: 1, 3, 5, 7, 9, ...
natuals = itertools.count(1)
# step 2: 取该序列的前N项: 1, 3, 5, 7, 9, ..., 2*N-1.
ns = itertools.takewhile(lambda x: x < N+1, natuals)
l = list(ns)
# step 3: 添加正负符号并用4除: 4/1, -4/3, 4/5, -4/7, 4/9, ...
for i in range(len(l)):
l[i] = 2 * l[i] - 1
if (i % 2 == 1):
l[i] = 0 - l[i]
l[i] = 4 / l[i]
# step 4: 求和:
return sum(l)
# 测试:
print(pi(10))
print(pi(100))
print(pi(1000))
print(pi(10000))
assert 3.04 < pi(10) < 3.05
assert 3.13 < pi(100) < 3.14
assert 3.140 < pi(1000) < 3.141
assert 3.1414 < pi(10000) < 3.1415
print('ok')