Householder similarity transformation of matrix

下边资料是关于Householder similarity transformation of matrix in Python的内容，希望能对小伙伴们有较大帮助。
''' d,c = householder(a).
Householder similarity transformation of matrix [a] to
the tridiagonal form [cdc].

p = computeP(a).
Computes the acccumulated transformation matrix [p]
after calling householder(a).

'''
from numpy import dot,diagonal,outer,identity
from math import sqrt

def householder(a):
n = len(a)
for k in range(n-2):
u = a[k+1:n,k]
uMag = sqrt(dot(u,u))
if u[0] < 0.0: uMag = -uMag
u[0] = u[0] + uMag
h = dot(u,u)/2.0
v = dot(a[k+1:n,k+1:n],u)/h
a[k+1:n,k+1:n] = a[k+1:n,k+1:n] - outer(v,u)

• outer(u,v)
  a[k,k+1] = -uMag
  return diagonal(a),diagonal(a,1)

def computeP(a):
n = len(a)
for k in range(n-2):
u = a[k+1:n,k]
h = dot(u,u)/2.0
v = dot(p[1:n,k+1:n],u)/h
p[1:n,k+1:n] = p[1:n,k+1:n] - outer(v,u)
return p