Python 数列のリストを作成する「Sympy」

　本記事では、等差数列や等比数列のリストを作成する雛形コードを載せました。下図上のリストは、初項1で交差5の等差数列、下図下のリストは、初項1、公比3の等比数列です。

■本プログラム

# -*- coding: utf-8 -*- 
#!/usr/bin/env python
from sympy import sequence
from sympy.abc import n

# 等差数列のリストを返す関数
def arithmetic_sequence(a0, d):
    myformula = sequence((a0+(n-1)*d), (n, start_n, end_n))
    return [int(i) for i in myformula]

# 等比数列のリストを返す関数
def geometric_progression(a0, r):
    myformula = sequence((a0*pow(r, n-1)), (n, start_n, end_n))
    return [int(i) for i in myformula]
    
def main():
    # 等差数列A
    a0=1 # 初項
    d=5 # 交差
    A_list = arithmetic_sequence(a0, d)
    print(A_list)

    # 等比数列B
    a0 = 1 # 初項
    r = 3 # 公比
    B_list = geometric_progression(a0, r)
    print(B_list)

if __name__ == '__main__':
    start_n = 1
    end_n = 10
    
    main()
    print('finished')

以上

<広告>

リンク