def main():
#print sys.argv
f = importer("ru.csv")
x, y =linearRegression(f[1], f[2], 0., 0., 0.000000001, 50)
plt.xlabel("Hora de Chegada")
plt.ylabel("Tempo de espera (minutos)")
plt.plot(f[1],f[2],'o',color='white',markersize=7,linewidth=3)
plt.plot(x, y,'k-')
plt.show()
for k in d:
s.append(loess_point(int(k),smoothness,d,e))
return d,np.array(s,dtype=float)
def loess_point(x,h,xp,yp):
w=exp(-0.5*(((x-xp)/h)**2)/sqrt(2*pi*h**2))
b=sum(w*xp)*sum(w*yp) - sum(w)*sum(w*xp*yp)
b /= sum(w*xp)**2 - sum(w)*sum(x*xp**2)
a=(sum(w*yp)-b*sum(w*xp))/sum(w)
return a+b*x
#d=loadtxt("ex.csv",delimiter=',')
ru=importer("ru.csv")
d=np.array(ru[1],dtype=float)
e=np.array(ru[2],dtype=float)
d,s1=loess(ru[1],ru[2],5)
xlabel("Day in Year")
ylabel("Draft Number")
plot(d,e,'o',color='white',markersize=7,linewidth=3)
plot(d,s1,'k-')
#q=4
#axis([1-q,366+q,1-q,366+q])
show()