Commit 225d9e24 authored by “Sivakorn”'s avatar “Sivakorn”

add function

parent 261d1ddb
เขียนฟังก์ชั่นเพื่อหา Upper-triangular matrix (U) จากสมการ Ax =b โดย Ux = c เป็นสมการที่ลดรูปแล้ว เขียนฟังก์ชั่นเพื่อหา Upper-triangular matrix (U) จากสมการ Ax =b โดย Ux = c เป็นสมการที่ลดรูปแล้ว
## How To Use
```
from hw06 import Ab2Uc;
import numpy as np
A = np.array([
[4, -2, 1],
[-2, 4, -2],
[1, -2, 4]], float)
b = np.array([11,-16,17], float)
Ab2Uc(A, b)
```
## Result
```
U : [[ 4. -2. 1. ]
[ 0. 3. -1.5]
[ 0. 0. 3. ]]
c : [ 11. -10.5 9. ]
```
\ No newline at end of file
def Ab2Uc(A,b):
n = len(b)
for j in range(0, n-1):
for i in range(j+1, n):
lam = A[i,j]/A[j,j]
A[i,j:n] = A[i, j:n] - lam*A[j, j:n]
b[i] = b[i] - lam*b[j]
print('U :',A)
print('c : ',b)
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment