FilesExpand file tree

master

/

linalg.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

21 lines (17 loc) · 443 Bytes

master

/

linalg.py

Copy path

Top

File metadata and controls

• Code
• Blame

21 lines (17 loc) · 443 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

from numpy import *

# import scipy

import matplotlib.pyplot as plt

c1,c2= 5.0,2.0

i = r_[1:11]

xi = 0.1*i

yi = c1*exp(-xi)+c2*xi

zi = yi + 0.05*max(yi)*random.randn(len(yi))

A = c_[exp(-xi)[:,newaxis],xi[:,newaxis]]

c,resid,rank,sigma = linalg.lstsq(A,zi)

xi2 = r_[0.1:1.0:100j]

yi2 = c[0]*exp(-xi2) + c[1]*xi2

plt.plot(xi,zi,'x',xi2,yi2)

plt.axis([0,1.1,3.0,5.5])

plt.xlabel('$x_i$')

plt.title('Data fitting with linalg.lstsq')

plt.show()