我正在编写一个必须解决某些递归关系的应用程序,但是有些问题无法通过使用rsolve
的{{1}}方法来解析解决。它只返回sympy
。有没有一种方法可以强制sympy对其进行数值求解?
我有这样的东西:
None
我希望有人能帮助我!
答案 0 :(得分:2)
这是我通过数值评估递归关系的解决方案。
当指定递归关系作为输入来简化时,请确保
值f(x)
不是字符串的一部分。也就是说,如果您的复发关系是
f(x) = -2*f(x-1)+11*f(x-2)+12*f(x-3)-36*f(x-4) +41**(x-4)+3
您的输入字符串应为:
"-2*f(x-1)+11*f(x-2)+12*f(x-3)-36*f(x-4) +41**(x-4)+3"
此外,此解决方案仅限于线性递归关系,尽管它也可以扩展到其他情况。
代码要做的是遍历递归关系的语法树,并通过数值计算或查找已知的f(x)值来评估每个节点。
from sympy import *
import operator
ctx = {
"f": Function("f"),
"x": var("x",integer=True)
}
initial_conditions = {
0: 1,
1: 1,
2: 1,
3: 1
}
func1 = sympify("-2*f(x-1)+11*f(x-2)+12*f(x-3)-36*f(x-4) +41**(x-4)+3", ctx)
def contains_function(f):
if issubclass(type(f),Function):
return True
r = map(contains_function,f.args)
return (sum(r) != 0)
def get_numeric_value(arg):
if arg.is_number:
if arg.is_integer:
return int(arg)
else:
return float(arg)
else:
return None
def evaluate_at(f, n, initial_conditions):
if f.is_Add:
result = 0
op = operator.add
elif f.is_Mul:
result = 1
op = operator.mul
elif f.is_Function:
func_arg = f.args[0]
func_arg_val = int(func_arg.subs(func_arg.free_symbols.pop(),n))
if not func_arg_val in initial_conditions:
return None
else:
return initial_conditions[func_arg_val]
else:
return None
for arg in f.args:
if arg.is_number:
result= op(result, get_numeric_value(arg))
elif contains_function(arg):
r = evaluate_at(arg,n,initial_conditions)
if r:
result=op(result,r)
else:
return None
else:
result =op(result,get_numeric_value(arg.subs(arg.free_symbols.pop(),n)))
return result
known_values = dict(initial_conditions)
for n in range(4,11):
known_values[n] = evaluate_at(func1,n,known_values)
答案 1 :(得分:0)
在这种情况下可能没有解析解决方案(见下文),我会考虑采用线性代数方法here:
from sympy import Function, rsolve
from sympy.abc import n
y = Function("y")
initial_conditions = {y(0): 1, y(1): 1, y(2): 1, y(3): 1}
f = -2*y(x-1)+11*y(x-2)+12*y(x-3)-36*y(x-4)+41**(x-4)+3-y(x)
print(rsolve(f, y(x)))
print(rsolve(f, y(x), initial_conditions))
返回:
(-3)**x*(C0 + C1*x) + 2**x*C1*(C0 + C1*x)/C0 + 41**x/2944656 + 3/16
None