数据分析学习记录

numpy之polyfit和poly1d

// An highlighted block
#引入包
import numpy as np
import matplotlib.pyplot as plt
#初始数据
x=[1,2,3,4,5,6,7,8]
y=[2.3,3.6,4.7,5.1,6.3,7.4,8.5,9.6]
#拟合数据一次多项式
z=np.polyfit(x,y,1)
#生成一次多项式
z1=np.poly1d(z)
print(z)
print(z1)
#模拟值
X=np.arange(1,9)
Y=z[0]*X+z[1]
plt.figure()
#初始值的散点图
plt.scatter(x,y)
plt.plot(X,Y)
plt.show()

注：如果使用x得到模拟值会报错，相乘的变量类型不一样。