Python - 数学 - SymPyで代数と式を計算する(級数の和)

Pythonからはじめる数学入門
原著

開発環境

• OS X El Capitan - Apple (OS)
• Emacs (Text Editor)
• Python 3.5 (プログラミング言語)

Pythonからはじめる数学入門 (Amit Saha (著)、黒川 利明 (翻訳)、オライリージャパン)の4章(SymPyで代数と式を計算する)、4.6(プログラミングチャレンジ)、問題4-3(級数の和)を取り組んでみる。

問題4-3(級数の和)

コード(Emacs)

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import sympy


def series(term, num):
    n = sympy.Symbol('n')
    s = sympy.summation(term, (n, 1, num))
    return s

if __name__ == '__main__':
    term = input('Enter the nth term: ')
    num = input('Enter the number of terms: ')

    try:
        term = sympy.sympify(term)
        num = int(num)
    except sympy.SympifyError as err:
        print(err)
    except Exception as err:
        print(err)
    else:
        sympy.pprint(series(term, num))

入出力結果(Terminal, IPython)

$ ./sample3.py
Enter the nth term: a + (n - 1) * d
Enter the number of terms: 3
3⋅a + 3⋅d
$