如何在去掉空数据及如何在matplotlib中显示数据线性拟合

来源:互联网 发布:淘宝乔丹是真的吗 编辑:程序博客网 时间:2024/04/29 15:56
import numpy as np
import scipy as sp
import matplotlib.pyplot as plt
def error(f,x,y):
    return sp.sum((f(x)-y)**2)


data = sp.genfromtxt('web_traffic.tsv',delimiter='\t')  
x = data[:,0]
y = data[:,1]
x = x[~sp.isnan(y)]   
y = y[~sp.isnan(y)]


plt.scatter(x,y)
plt.title('Web traffic over the last month')
plt.xlabel('Time')
plt.ylabel('Hits/hour')
plt.xticks([w*7*24 for w in range(10)],['week %i' %w for w in range(10)])
plt.autoscale(tight = True)
plt.grid()
fp1,residuals,rank,sv,rcond= sp.polyfit(x,y,1,full=True)
print("残差:",residuals)
print('Model parameter:',fp1)
f1 = sp.poly1d(fp1)
print(error(f1,x,y))
fx = sp.linspace(0,x[-1],1000)
plt.plot(fx,f1(fx),linewidth=4,color='red')
plt.legend(['d=%i' %f1.order],loc = 'upper left')
plt.show()
阅读全文
0 0