19

次の関数を想定します。

f(x) = x * cos(x-4)

この関数を使用すると、とでx = [-2.5, 2.5]交差0します。f(0) = 0f(-0.71238898) = 0

これは、次のコードで決定されました。

import math
from scipy.optimize import fsolve
def func(x):
    return x*math.cos(x-4)
x0 = fsolve(func, 0.0)
# returns [0.]
x0 = fsolve(func, -0.75)
# returns [-0.71238898]

fzero1回の呼び出しで両方のルートを見つけるために(または他のPythonルートファインダー)を使用する適切な方法は何ですか?これを行う別のscipy機能はありますか?

fzero参照

4

3 に答える 3

19

私はかつてこのタスクのモジュールを作成しました。これは、 JaanKiusalaas著の『Numerical Methods in Engineering with Python』の4.3章に基づいています。

import math

def rootsearch(f,a,b,dx):
    x1 = a; f1 = f(a)
    x2 = a + dx; f2 = f(x2)
    while f1*f2 > 0.0:
        if x1 >= b:
            return None,None
        x1 = x2; f1 = f2
        x2 = x1 + dx; f2 = f(x2)
    return x1,x2

def bisect(f,x1,x2,switch=0,epsilon=1.0e-9):
    f1 = f(x1)
    if f1 == 0.0:
        return x1
    f2 = f(x2)
    if f2 == 0.0:
        return x2
    if f1*f2 > 0.0:
        print('Root is not bracketed')
        return None
    n = int(math.ceil(math.log(abs(x2 - x1)/epsilon)/math.log(2.0)))
    for i in range(n):
        x3 = 0.5*(x1 + x2); f3 = f(x3)
        if (switch == 1) and (abs(f3) >abs(f1)) and (abs(f3) > abs(f2)):
            return None
        if f3 == 0.0:
            return x3
        if f2*f3 < 0.0:
            x1 = x3
            f1 = f3
        else:
            x2 =x3
            f2 = f3
    return (x1 + x2)/2.0

def roots(f, a, b, eps=1e-6):
    print ('The roots on the interval [%f, %f] are:' % (a,b))
    while 1:
        x1,x2 = rootsearch(f,a,b,eps)
        if x1 != None:
            a = x2
            root = bisect(f,x1,x2,1)
            if root != None:
                pass
                print (round(root,-int(math.log(eps, 10))))
        else:
            print ('\nDone')
            break

f=lambda x:x*math.cos(x-4)
roots(f, -3, 3)

rootsf区間[ ab]内ののすべての根を見つけます。

于 2012-10-24T18:54:30.193 に答える
13

スカラー配列またはnumpy配列のいずれかを引数として取ることができるように、関数を定義します。

>>> import numpy as np
>>> f = lambda x : x * np.cos(x-4)

次に、引数のベクトルをに渡しますfsolve

>>> x = np.array([0.0, -0.75])
>>> fsolve(f,x)
array([ 0.        , -0.71238898])
于 2012-10-24T18:44:48.023 に答える
4

一般に(つまり、関数が特定のクラスに属していない限り)、すべてのグローバルソリューションを見つけることはできません。これらのメソッドは通常、指定された開始点からローカル最適化を実行します。

ただし、math.cos()をnumpy.cos()に切り替えることができます。これにより、関数がベクトル化されるため、fsolve(func、np.arange(-10,10,0.5))などの多くの値を一度に解決できます。

于 2012-10-24T18:45:08.273 に答える