script/cpython/linfit.py

@@ -0,0 +1,32 @@
+import numpy as np
+
+def linfit(x, y):
+    """
+    Return linear fit
+    """
+    p = np.polyfit(x, y, 1)
+    f = np.poly1d(p)
+    x_fit = np.linspace(min(x), max(x), 100)
+    y_fit = f(x_fit)
+    yhat = f(x)
+    ybar = np.sum(y)/len(y)
+    ssreg = np.sum((yhat - ybar)**2)
+    sstot = np.sum((y - ybar)**2)
+    R2 = ssreg / sstot
+    return (p, x_fit, y_fit, R2)
+
+def test():
+    return np.ones(5)
+
+
+def test2(name, x=None, y=None):
+    print (name,x,y)
+    ret = y*x  
+    print (ret)
+    return ret
+
+def add(x,y,z):
+    return x+y+z
+
+def read_dev(dev):
+    return dev.read()