廖雪峰课后习题【itertools】python计算圆周率

课后习题，根据提示写出来的程序，对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')

转载于:https://my.oschina.net/DBboy/blog/1801408