fn 实现无限计算等比级数

今天看InfoQ上的一篇关于Fn.py的文章，Fn中的Stream可以实现无限序列，例如文章中的代码：

f = Stream()
fib = f << [0, 1] << map(lambda x, y: x + y, f, drop(1, f))

这是实现了一个无限的斐波那契数列，我根据这个实现了一个等比级数，等比级数的公式是f = b^(n-1), b是基数，n=1 f=1, n>=2 f=b^(n-1)。

我实现的方法使用了生成器：

def geo_sequence(b):
    n = 2
    while 1:
        yield b ** (n - 1)
        n += 1

gs = geo_sequence(3)
print gs.next() # output 3
print gs.next() # output 9
print gs.next() # output 27
print gs.next() # output 81

f = Stream()
f << geo_sequence(3)
print f[3]
print list(f[:5])
print list(f[3:5])
print list(f[5:10])

81
[3, 9, 27, 81, 243]
[81, 243]
[729, 2187, 6561, 19683, 59049]

就这样！