[SciPy-User] fitting multiple coefficients

Keith Goodman kwgoodman at gmail.com
Sun Jul 25 12:59:12 EDT 2010


On Sun, Jul 25, 2010 at 9:48 AM, eneide.odissea
<eneide.odissea at gmail.com> wrote:
> Hi all
> I have the following function
> y = m[0] A + m[1] B + m[2] C
> and I have the following data
> A   ;   B  ;     C   ;    y
> 10    ;  122 ;  7   ;    4
> 12    ;  134 ;  9   ;    2
> 11    ;  131 ;  6   ;    4
> 7      ;  180 ;  4   ;    2
> Do you know what function of python I could use to find
> the weights m[i] that  best fit for these data?

>> import numpy as np
>> x = np.array([[10, 122, 7], [12, 134, 9], [11, 131, 6], [7, 180, 4]])
>> y = np.array([4, 2, 4, 2])
>> m, ig1, ig2, ig3 = np.linalg.lstsq(x, y)
>> m
   array([ 0.84562416, -0.00455408, -0.74598866])
>> np.dot(x, m) # <-- fitted y
   array([ 2.67872369,  2.82334576,  4.22934983,  2.11568083])



More information about the SciPy-User mailing list