#!/usr/bin/env python
# coding: utf-8
# # Table of Contents
#
#
#
#
# 線形代数-逆行列
#
#
#
# cc by Shigeto R. Nishitani 2017-19
#
#
#
#
#
# # 行列計算の概要
#
#
# 数値計算の中心課題の一つである，行列に関する演算について見ていきます．
# 多次元，大規模な行列に対する効率のよい計算法が多数開発されており，
# 多くの既存のライブラリが用意されています．
# 本章ではそれらの中心をなす，
# 逆行列(matrix inverse)と固有値(Eigen values)に関して具体的な計算方法を示します．
# 現実的な問題には既存のライブラリを使うのが上策ですが，
# それでも基礎となる原理の理解や，ちょっとした計算，
# ライブラリの結果の検証に使えるルーチンを示します．
#
# 逆行列は連立一次方程式を解くことと等価です．
# ルーチン的なやり方にガウスの消去法があります．
# これは上三角行列になれば代入を適宜おこなうことで
# 解が容易に求まることを利用します．
# さらに，初期値から始めて次々に解に近づけていく反復法があります．
# この代表例であるJacobi(ヤコビ)法と，
# 収束性を高めたGauss-Seidel(ガウス-ザイデル)法を紹介します．
#
# 上記の手法をより高速にした修正コレスキー分解と共役傾斜(共役勾配)法がありますが，
# 複雑になるのでパスります．
# 必要ならばNumRecipeを読んでください．
#
#
# # ガウス消去法による連立一次方程式の解
#
# 逆行列は連立一次方程式を解くことと等価です．
# すなわち，$A$を行列，$x$を未知数ベクトル，$b$を数値ベクトルとすると，
#
# $$
# \begin{array}{rl} Ax &= b \\
# A^{-1}Ax &= A^{-1}b \\
# x &= A^{-1}b
# \end{array}
# $$
# です．
# 未知数の少ない連立一次方程式では，適当に組み合わせて未知数を消していけばいいですが，
# 未知数が多くなってしまうと破綻します．
# 未知数の多い多元連立一次方程式で，ルーチン的に解を求めていく方法がガウス消去法で，
# **前進消去**と**後退代入**という2つの操作からなります．
# 後退代入(Backward substitution)による解の求め方を先ず見ましょう．
# たとえば，
#
# $$
# \begin{array}{rl}
# x+y-2z & = -4 \\
# -3y+3z & = 9\\
# -2z & = -3
# \end{array}
# $$
# では，下から順番に$z\rightarrow y\rightarrow x$と適宜代入することによって，
# 簡単に解を求めることが出来ます．
# 係数で作る行列でこのような形をした上三角行列にする操作を
# 前進消去あるいはガウスの消去法(Gaussian elimination)といいます．
# 下三角行列L(lower triangular matrix)と上三角行列U(upper triangular matrix)の積に分解する操作
#
# $$
# A = L.U
# $$
# をLU分解(LU decomposition)といいます．
# 例えば先に示した上三角行列を係数とする連立方程式は，
#
# $$
# \begin{array}{rl}
# x+y-2z&=-4 \\
# x-2y+z&=5 \\
# x-2y-z&=2
# \end{array}
# $$
# を変形することで得られます．連立方程式からの変形を示してください．
#
# | |
# |:----|
#
#
#
# ```
#
# ```
# # pythonによるLU分解
#
# 係数行列(coefficient matrix)Aと定数項(b)を定義します．
# In[1]:
from pprint import pprint
import scipy.linalg as linalg # SciPy Linear Algebra Library
import numpy as np
# A = np.array([[1,1,-2,-4],[1,-2,1,5],[2,-2,-1,2]])
A = np.array([[1,1,-2],[1,-2,1],[1,-2,-1]])
b = np.array([[-4],[5],[2]])
pprint(A)
pprint(b)
# 単に逆行列を求める際は
# In[2]:
inv_A = linalg.inv(A)
print(inv_A)
np.dot(inv_A,b)
# 拡大係数行列はhstack(horizontal stack)かcolumn_stackを使います．
# In[3]:
e_A = np.hstack((A, b))
pprint(e_A)
pprint(np.column_stack((A,b)))
# In[4]:
P, L, U = linalg.lu(e_A)
print("P:")
pprint(P)
print("L:")
pprint(L)
print("U:")
pprint(U)
# 元の拡大行列はこれらの掛け算で求められます．
# In[5]:
print("e_A=P.L.U:")
pprint(np.dot(np.dot(P,L),U))
# # LU分解のコード
#
# LU分解すれば線形方程式の解が容易に求まることは理解できると思う．具体的に$A$をLU分解する行列(消去行列と称す)T1,T2の係数は次のようにして求められる．
# In[6]:
from pprint import pprint
import scipy.linalg as linalg # SciPy Linear Algebra Library
import numpy as np
A = np.array([[1.0,1,-2],[1,-2,1],[1,-2,-1]])
A0 = np.array([[1.0,1,-2],[1,-2,1],[1,-2,-1]])
b = np.array([[-4.0],[5],[2]])
n = 3
L = np.identity(n)
T = []
for i in range(n): #i行目
T.append(np.identity(n))
for j in range(i+1, n):
am = A[j,i]/A[i,i] #i行の要素を使って，i+1行目の先頭を消す係数を求める
T[i][j,i]=-am #i番目の消去行列に要素を入れる
L[j,i]=am #LTMの要素
for k in range(n):
A[j,k] -= am*A[i,k] #もとの行列をUTMにしていく
b[j] -= b[i]*am #数値ベクトルも操作
pprint(A)
pprint(b)
# 上のコードによって得られた消去行列．
# In[7]:
pprint(T)
# これを実際に元の行列$A0$に作用させると，UTMが求められる．
# In[8]:
np.dot(T[1],np.dot(T[0],A0))
# 求められたLTM, UTMを掛けると
# In[9]:
np.dot(L,A)
# 元の行列を得られる．L,Aに求めたい行列が入っていることを確認．
# In[10]:
pprint(L)
pprint(A)
# 数値ベクトルも期待通り変換されている．
# In[11]:
pprint(b)
# luの出力で得られる拡張係数行列のUTMは，次の通り．
# In[12]:
np.hstack((A, b))
# # ピボット操作
#
#
# ガウス消去法で困るのは，割ろうとした対角要素が0の場合である．しかし，この場合にも，方程式の順序を，行列の行と右辺の値をペアにして入れ替えれば解決する．この割る
# ほうの要素をピボット要素あるいはピボット(pivot,バスケの軸足を動かさずにくるくる回すやつ)と呼ぶ．この操作は，変数の並びを変えたわけではなく，単に方程式の
# 順番を変更する操作に相当する．
#
# さらに対角要素の数値が厳密に0でなくとも，極端に0に近づいた場合にも，その数で割った数値が大きくなり他の数との差を取ると以前に示した情報落ちの可能性が出てくる．
# この現象を防ぐためには，絶対値が最大のピボットを選んで行の入れ替えを毎回おこなうといい結果が得られることが知られている．
#
# MapleのLUDecompositionコマンドをこのような行列に適用すると，置換行列(permutation
# matrix)Pが単位行列ではなく，ピボット操作に対応した行列となる．P.A=L.Uとなることに注意．
#
#
# 下の例では，$a_{11}$と$a_{31}$の数値を少し変えている．
# その結果，置換行列Pはidentity行列ではなく，
# 2，3行目でpivotが起こっていることを表している．
# In[13]:
A = np.array([[2,1,-2],[1,-2,1],[2,-2,-1]])
b = np.array([[-4.0],[5],[2]])
e_A = np.hstack((A, b))
pprint(e_A)
P, L, U = linalg.lu(e_A)
print("P:")
pprint(P)
print("L:")
pprint(L)
print("U:")
pprint(U)
# # 反復法による連立方程式の解
#
#
# 以下のような連立方程式を
#
# $$
# \left[ \begin {array}{c} 5\,x+y+z+u\\ x+3\,y+z+u\\ x-2\,y-9\,z+u\\ x+3\,y-2\,z+5\,u\end {array} \right] = \left[ \begin {array}{c} -6\\ 2\\ -7\\ 3\end {array} \right]
# $$
# 形式的に解くと
#
# $$
# x=\frac{-6-(y+z+u)}{5}
# $$
# となる．他の未知数も，
#
# |y=| |
# |:----|:----|
# |z=| |
# |u=| |
#
# となる．適当に初期値($x_0,y_0,z_0,u_0$)をとり，下側の方程式に代入すると，得られた出力($x_1,y_1,z_1,u_1$)はより精解に近い値となる．これを繰り返すことによって精解が得られる．これをヤコビ(Jacobi)法と呼び，係数行列の対角要素が非対角要素にくらべて大きいときに適用できる．多くの現実の問題ではこの状況が成り立っている．
#
# Gauss-Seidel法はJacobi法の高速版である．$n$番目の解の組が得られた後に一度に次の解の組に入れ替えるのではなく，得られた解を順次改良した解として使っていく．これにより，収束が早まる．以下にはヤコビ法のコードを示した．x1[i]の配列を変数に換えるだけで，Gauss-Seidel法となる．
#
# In[14]:
# Jacobi iterative method for solving Ax=b by bob
import numpy as np
np.set_printoptions(precision=6, suppress=True)
A=np.array([[5,1,1,1],[1,3,1,1],[1,-2,-9,1],[1,3,-2,5]])
b=np.array([-6,2,-7,3])
n=4
x0=np.zeros(n)
x1=np.zeros(n)
for iter in range(0, 20):
for i in range(0, n):
x1[i]=b[i]
for j in range(0, n):
x1[i]=x1[i]-A[i][j]*x0[j]
x1[i]=x1[i]+A[i][i]*x0[i]
x1[i]=x1[i]/A[i][i]
for j in range(0, n):
x0[j]=x1[j]
print(x0)
# もう少しpython(numpy)の機能を使うとおしゃれな書き方ができる．
# 以下は，ネットから拾ってきたcode.
# 上(by bob)とやってることの本質的なところは同じですが，
# numpyのindexを自動的に判断する機能を使って簡潔に書いています．
# こういうのを読める，書けるようになれば，あなたもpython使い．
# 私はまだpythonに馴染めてなくて，いちいち頭の中で治しています．．．
# In[15]:
# https://www.quantstart.com/articles/Jacobi-Method-in-Python-and-NumPy
from pprint import pprint
from numpy import array, zeros, diag, diagflat, dot
np.set_printoptions(precision=6, suppress=True)
def jacobi(A,b,N=25,x=None):
if x is None:
x = zeros(len(A[0]))
D = diag(A)
R = A - diagflat(D)
# Iterate for N times
for i in range(N):
x = (b - dot(R,x)) / D
print(x)
return x
A = array([[5,1,1,1],[1,3,1,1],[1,-2,-9,1],[1,3,-2,5]])
b = array([-6,2,-7,3])
sol = jacobi(A,b,N=10)
# # 課題
#
# ## 行列AをLU分解
# 次の行列AをLU分解し，上・下三角行列を求めよ．さらに連立方程式の解を求めよ.
# $$
# \left[
# \begin {array}{c} x+4\,y+3\,z\\
# x-2\,y+z \\
# 2\,x-2\,y-z
# \end {array} \right] =
# \left[
# \begin {array}{c} 11\\
# 11\\
# 11
# \end {array} \right]
# $$
# ## pivot付きのLU分解
#
# 次の連立方程式の拡大行列をLU分解せよ．pivot操作が必要となる．
# P.A = L.Uを確かめよ．
# $$
# \left[ \begin {array}{c} 3\,w+2\,x+2\,y+z\\ 3\,w+2\,x+3\,y+z\\ w-2\,x-3\,y+z\\ 5\,w+3\,x-2\,y+5\,z\end {array} \right] = \left[ \begin {array}{c} -6\\ 2\\ -9\\ 2\end {array} \right]
# $$
# ## Gauss-Seidel法
# Jacobi法のプログラムを参照してGauss-Seidel法のプログラムを作れ．
# ```
# A=np.array([[5,1,1,1],[1,3,1,1],[1,-2,-9,1],[1,3,-2,5]])
# b=np.array([-6,2,-7,3])
# n=4
# ```
# に対して計算を行い，Jacobi法と収束性を比べよ．
#
# 最初の数回はこんな感じ．
#
# ```
# [-1.2 1.066667 0.407407 0.362963]
# [-1.567407 0.932346 0.436763 0.528779]
# [-1.579578 0.871345 0.46739 0.580064]
# [-1.58376 0.845435 0.478382 0.600844]
# [-1.584932 0.835236 0.482827 0.608976]
# ```
# In[ ]: