Вот мое решение для численной оценки рекуррентных отношений.Когда вы указываете отношение повторения в качестве входных данных для упрощения, убедитесь, что значение 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)