Pythonで総和を実装

総和の式を実装してみる

\sum_{i=m}^{n} f(x) = f(m) + f(m+1) + \cdots + f(n) \\

sigma.py

def sigma(m, n, func, s = 0) :
    if m > n: return s
    return sigma(m + 1, n, func, s + func(m))

使用例

sigma_test.py

print(sigma(1, 10, lambda x : x))
55

print(sigma(1, 3, lambda x : 3 * 5 ** (x - 1)))
93

Webで適当な問題見つけて入力したけどちゃんと動いてそう。
もっと効率のいい書き方とか、もっとオサレな書き方とか、もうそれ用意されてるしとかもあるのかも。

追記

NumPyという数学関数ライブラリがあるので大抵の式は揃っていそう。。。

Pythonの数値計算ライブラリ NumPy入門
http://rest-term.com/archives/2999/