sympyで解いた連立方程式の解を変数に代入したい

Question

前提

Pythonista3
等差数列の問題を解くためのプログラムを作っています。

実現したいこと

他サイトの連立方程式の解き方を参考にしました。
sympyを使って連立方程式を解き、初項「a」と公差「d」を求めました。
連立方程式を解く所までしか知ることが出来なかったため質問させていただきます。

「第5項が28、第15項が−2である等差数列の初項と公差を求めよ。」
one = 5
one_n = 28
two = 15
two_n = -2
(max = 100)
を入力すると、

{初項: 40, 公差: -3}

と出力されます。

この解を変数に代入して
その後の一般項を求める計算などに使いたいです。

該当のソースコード

import sympy
def q():
	
	a = sympy.Symbol("初項")
	d = sympy.Symbol("公差")
	
	one = int(input("①第何項？ >>> "))
	one_n = int(input(f"第{one}項の数字を入力 >>> "))
	two = int(input("②第何項？ >>>"))
	two_n = int(input(f"第{two}項の数字を入力 >>> "))
	max = int(input("第何項まで知りたい？ >>> "))

	equation1 = a + (one-1)*d - (one_n)
	equation2 = a + (two-1)*d - (two_n)  
	
	print(sympy.solve([equation1,equation2]))

Answer 1

この解を変数に代入して
その後の一般項を求める計算などに使いたいです。

計算結果からそれぞれの値を取り出せば良いです。下記例ではfloat値に変換しています。

a = sympy.Symbol("初項")
d = sympy.Symbol("公差")

one = int(input("①第何項？ >>> "))
one_n = int(input(f"第{one}項の数字を入力 >>> "))
two = int(input("②第何項？ >>>"))
two_n = int(input(f"第{two}項の数字を入力 >>> "))
max = int(input("第何項まで知りたい？ >>> "))

equation1 = a + (one-1)*d - (one_n)
equation2 = a + (two-1)*d - (two_n)  

result = sympy.solve([equation1,equation2])

syokou = float(result[a])
kousa = float(result[d])
print(syokou, kousa)

Comments

[Plasma]

ありがとうございます。正しく処理ができました！