sympy不使用rsolve计算n次重复

时间:2018-12-12 18:05:29

标签: python sympy

我正在编写一个必须解决某些递归关系的应用程序,但是有些问题无法通过使用rsolve的{​​{1}}方法来解析解决。它只返回sympy。有没有一种方法可以强制sympy对其进行数值求解?

我有这样的东西:

None

我希望有人能帮助我!

2 个答案:

答案 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