scipy.linalg.

qr#

scipy.linalg.qr(a, overwrite_a=False, lwork=None, mode='full', pivoting=False, check_finite=True)[源代码]#

计算矩阵的 QR 分解。

计算分解 A = Q R,其中 Q 是酉/正交矩阵,R 是上三角矩阵。

参数
a(M, N) 类数组

要分解的矩阵

overwrite_abool,可选

是否覆盖 a 中的数据(如果将 overwrite_a 设置为 True,通过重用现有的输入数据结构而不是创建新的数据结构,可能会提高性能。)

lworkint,可选

工作数组大小,lwork >= a.shape[1]。如果为 None 或 -1,则计算最佳大小。

mode{‘full’, ‘r’, ‘economic’, ‘raw’},可选

确定要返回的信息:Q 和 R(‘full’,默认),仅 R(‘r’),或 Q 和 R 但以经济尺寸计算(‘economic’,参见注释)。最后一个选项 ‘raw’(在 SciPy 0.11 中添加)使该函数返回 LAPACK 使用的内部格式的两个矩阵 (Q, TAU)。

pivotingbool,可选

分解是否应包含用于揭示秩的 qr 分解的旋转。如果旋转,则计算分解 A[:, P] = Q @ R,如上所述,但其中选择 P 使得 R 的对角线是不递增的。等效地,尽管效率较低,但可以通过排列单位矩阵的行或列(取决于它要使用的方程的哪一侧)来显式形成显式的 P 矩阵。请参阅示例。

check_finitebool,可选

是否检查输入矩阵是否仅包含有限数字。禁用可能会提高性能,但如果输入包含无穷大或 NaN,则可能会导致问题(崩溃,不终止)。

返回
Q浮点数或复数 ndarray

形状为 (M, M),对于 mode='economic',形状为 (M, K)。如果 mode='r',则不返回。如果 mode='raw',则替换为元组 (Q, TAU)

R浮点数或复数 ndarray

形状为 (M, N),对于 mode in ['economic', 'raw'],形状为 (K, N)。K = min(M, N)

P整数 ndarray

对于 pivoting=True,形状为 (N,)。如果 pivoting=False,则不返回。

引发
LinAlgError

如果分解失败则引发

注释

这是 LAPACK 例程 dgeqrf、zgeqrf、dorgqr、zungqr、dgeqp3 和 zgeqp3 的接口。

如果 mode=economic,则 Q 和 R 的形状分别为 (M, K) 和 (K, N),而不是 (M,M) 和 (M,N),其中 K=min(M,N)

示例

>>> import numpy as np
>>> from scipy import linalg
>>> rng = np.random.default_rng()
>>> a = rng.standard_normal((9, 6))
>>> q, r = linalg.qr(a)
>>> np.allclose(a, np.dot(q, r))
True
>>> q.shape, r.shape
((9, 9), (9, 6))
>>> r2 = linalg.qr(a, mode='r')
>>> np.allclose(r, r2)
True
>>> q3, r3 = linalg.qr(a, mode='economic')
>>> q3.shape, r3.shape
((9, 6), (6, 6))
>>> q4, r4, p4 = linalg.qr(a, pivoting=True)
>>> d = np.abs(np.diag(r4))
>>> np.all(d[1:] <= d[:-1])
True
>>> np.allclose(a[:, p4], np.dot(q4, r4))
True
>>> P = np.eye(p4.size)[p4]
>>> np.allclose(a, np.dot(q4, r4) @ P)
True
>>> np.allclose(a @ P.T, np.dot(q4, r4))
True
>>> q4.shape, r4.shape, p4.shape
((9, 9), (9, 6), (6,))
>>> q5, r5, p5 = linalg.qr(a, mode='economic', pivoting=True)
>>> q5.shape, r5.shape, p5.shape
((9, 6), (6, 6), (6,))
>>> P = np.eye(6)[:, p5]
>>> np.allclose(a @ P, np.dot(q5, r5))
True