mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2025-03-28 09:26:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2024-11-20 09:16:12 +00:00
parent ab246373e6
commit d470987ca8
1 changed files with 15 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
linear_algebra/Lanczos-algorithm.py
View File

@ -1,20 +1,22 @@
import numpy as np
def lanczos(A: np.ndarray) -> ([float], [float]):
"""
Implements the Lanczos algorithm for a symmetric matrix.
Implements the Lanczos algorithm for a symmetric matrix.
Parameters:
-----------
matrix : numpy.ndarray
Symmetric matrix of size (n, n).
Parameters:
-----------
matrix : numpy.ndarray
Symmetric matrix of size (n, n).
Returns:
--------
alpha : [float]
List of diagonal elements of the resulting tridiagonal matrix.
beta : [float]
List of off-diagonal elements of the resulting tridiagonal matrix.
"""
Returns:
--------
alpha : [float]
List of diagonal elements of the resulting tridiagonal matrix.
beta : [float]
List of off-diagonal elements of the resulting tridiagonal matrix.
"""
n = A.shape[0]
V = np.zeros((n, n))
V[:, 0] = np.random.randn(n)
@ -31,4 +33,4 @@ def lanczos(A: np.ndarray) -> ([float], [float]):
w -= beta[j - 1] * V[:, j - 1]
beta.append(np.linalg.norm(w))
V[:, j + 1] = w / beta[j]
return alpha, beta
return alpha, beta