Python SciPy linalg.cho_solve_banded用法及代碼示例

本文簡要介紹 python 語言中 scipy.linalg.cho_solve_banded 的用法。

用法:

scipy.linalg.cho_solve_banded(cb_and_lower, b, overwrite_b=False, check_finite=True)#

求解線性方程 A x = b ，給定帶狀 Hermitian A 的 Cholesky 分解。

參數 ：：

(cb, lower)： 元組，(ndarray，布爾)

cb 是 A 的 Cholesky 因式分解，由 cholesky_banded 給出。 lower 必須與賦予cholesky_banded 的值相同。

b： array_like

右側

overwrite_b： 布爾型，可選

如果為 True，該函數將覆蓋 b 中的值。

check_finite： 布爾型，可選

是否檢查輸入矩陣是否僅包含有限數。禁用可能會提高性能，但如果輸入確實包含無窮大或 NaN，則可能會導致問題(崩潰、非終止)。

返回 ：：

x： 數組

係統 A x = b 的解

注意：

例子：

>>> import numpy as np
>>> from scipy.linalg import cholesky_banded, cho_solve_banded
>>> Ab = np.array([[0, 0, 1j, 2, 3j], [0, -1, -2, 3, 4], [9, 8, 7, 6, 9]])
>>> A = np.diag(Ab[0,2:], k=2) + np.diag(Ab[1,1:], k=1)
>>> A = A + A.conj().T + np.diag(Ab[2, :])
>>> c = cholesky_banded(Ab)
>>> x = cho_solve_banded((c, False), np.ones(5))
>>> np.allclose(A @ x - np.ones(5), np.zeros(5))
True