mirror of
https://hub.njuu.cf/TheAlgorithms/Python.git
synced 2023-10-11 13:06:12 +08:00
Changed the deprecated np.matrix
to np.ndarray
(#1923)
This commit is contained in:
parent
308505f18f
commit
bcaa88b26c
@ -4,56 +4,55 @@ https://en.wikipedia.org/wiki/Rayleigh_quotient
|
|||||||
import numpy as np
|
import numpy as np
|
||||||
|
|
||||||
|
|
||||||
def is_hermitian(matrix: np.matrix) -> bool:
|
def is_hermitian(matrix: np.array) -> bool:
|
||||||
"""
|
"""
|
||||||
Checks if a matrix is Hermitian.
|
Checks if a matrix is Hermitian.
|
||||||
|
|
||||||
>>> import numpy as np
|
>>> import numpy as np
|
||||||
>>> A = np.matrix([
|
>>> A = np.array([
|
||||||
... [2, 2+1j, 4],
|
... [2, 2+1j, 4],
|
||||||
... [2-1j, 3, 1j],
|
... [2-1j, 3, 1j],
|
||||||
... [4, -1j, 1]])
|
... [4, -1j, 1]])
|
||||||
>>> is_hermitian(A)
|
>>> is_hermitian(A)
|
||||||
True
|
True
|
||||||
>>> A = np.matrix([
|
>>> A = np.array([
|
||||||
... [2, 2+1j, 4+1j],
|
... [2, 2+1j, 4+1j],
|
||||||
... [2-1j, 3, 1j],
|
... [2-1j, 3, 1j],
|
||||||
... [4, -1j, 1]])
|
... [4, -1j, 1]])
|
||||||
>>> is_hermitian(A)
|
>>> is_hermitian(A)
|
||||||
False
|
False
|
||||||
"""
|
"""
|
||||||
return np.array_equal(matrix, matrix.H)
|
return np.array_equal(matrix, matrix.conjugate().T)
|
||||||
|
|
||||||
|
|
||||||
def rayleigh_quotient(A: np.matrix, v: np.matrix) -> float:
|
def rayleigh_quotient(A: np.array, v: np.array) -> float:
|
||||||
"""
|
"""
|
||||||
Returns the Rayleigh quotient of a Hermitian matrix A and
|
Returns the Rayleigh quotient of a Hermitian matrix A and
|
||||||
vector v.
|
vector v.
|
||||||
>>> import numpy as np
|
>>> import numpy as np
|
||||||
>>> A = np.matrix([
|
>>> A = np.array([
|
||||||
... [1, 2, 4],
|
... [1, 2, 4],
|
||||||
... [2, 3, -1],
|
... [2, 3, -1],
|
||||||
... [4, -1, 1]
|
... [4, -1, 1]
|
||||||
... ])
|
... ])
|
||||||
>>> v = np.matrix([
|
>>> v = np.array([
|
||||||
... [1],
|
... [1],
|
||||||
... [2],
|
... [2],
|
||||||
... [3]
|
... [3]
|
||||||
... ])
|
... ])
|
||||||
>>> rayleigh_quotient(A, v)
|
>>> rayleigh_quotient(A, v)
|
||||||
matrix([[3.]])
|
array([[3.]])
|
||||||
"""
|
"""
|
||||||
v_star = v.H
|
v_star = v.conjugate().T
|
||||||
return (v_star * A * v) / (v_star * v)
|
return (v_star.dot(A).dot(v)) / (v_star.dot(v))
|
||||||
|
|
||||||
|
|
||||||
def tests() -> None:
|
def tests() -> None:
|
||||||
A = np.matrix([[2, 2 + 1j, 4], [2 - 1j, 3, 1j], [4, -1j, 1]])
|
A = np.array([[2, 2 + 1j, 4], [2 - 1j, 3, 1j], [4, -1j, 1]])
|
||||||
v = np.matrix([[1], [2], [3]])
|
v = np.array([[1], [2], [3]])
|
||||||
assert is_hermitian(A), f"{A} is not hermitian."
|
assert is_hermitian(A), f"{A} is not hermitian."
|
||||||
print(rayleigh_quotient(A, v))
|
print(rayleigh_quotient(A, v))
|
||||||
|
|
||||||
A = np.matrix([[1, 2, 4], [2, 3, -1], [4, -1, 1]])
|
A = np.array([[1, 2, 4], [2, 3, -1], [4, -1, 1]])
|
||||||
assert is_hermitian(A), f"{A} is not hermitian."
|
assert is_hermitian(A), f"{A} is not hermitian."
|
||||||
assert rayleigh_quotient(A, v) == float(3)
|
assert rayleigh_quotient(A, v) == float(3)
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user