blob: 8fb88b290c2625667d0df434c320809733d39823 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
#!/usr/bin/python
#/usr/local/bin/python2.7
import numpy as np
def main():
dataz = np.genfromtxt('example.csv', delimiter=',', names=True)
y = dataz['attitude']
x = dataz['correct']
print 'x = {} {}'.format(x, np.mean(x))
print 'y = {} {}'.format(y, np.mean(y))
delta_x = x - np.mean(x)
delta_y = y - np.mean(y)
delta_xy = delta_x*delta_y
delta2_x = delta_x**2
delta2_y = delta_y**2
#print 'delta_x = {} {}'.format(delta_x, np.sum(delta_x))
#print 'delta_y = {} {}'.format(delta_y, np.sum(delta_y))
#print 'delta_xy = {} {}'.format(delta_xy, np.sum(delta_xy))
#print 'delta2_x = {} {}'.format(delta2_x, np.sum(delta2_x))
#print 'delta2_y = {} {}'.format(delta2_y, np.sum(delta2_y))
sx = np.sqrt(np.sum(delta2_x)/(delta2_x.size - 1))
sy = np.sqrt(np.sum(delta2_y)/(delta2_y.size - 1))
r = np.sum(delta_xy)/np.sqrt(np.sum(delta2_x)*np.sum(delta2_y))
#print 'sx = {}'.format(sx)
#print 'sy = {}'.format(sy)
print 'r = {}'.format(r)
print 'r^2 = {}'.format(r**2)
b = r*(sy/sx)
a = np.mean(y) - b*np.mean(x)
print 'b = {}'.format(b)
print 'a = {}'.format(a)
err = y - (a + b*x)
b_err = np.sqrt(np.sum(err**2.0)/((x.size - 2)*np.sum(delta2_x)))
a_err = np.sqrt(np.sum(x**2.0)*np.sum(err**2.0)/(x.size*(x.size - 2)*np.sum(delta2_x)))
print 'b_err = {}'.format(b_err)
print 'a_err = {}'.format(a_err)
print ' (attitude) \t= {}(correct) + {}'.format(b, a)
if __name__ == '__main__':
main()
|
