from numpy import genfromtxt
from sklearn import linear_model # 可以直接调用回归分析
dataPath = r"Delivery.csv"
deliveryData = genfromtxt(dataPath, delimiter=',') # 转化数据格式
print("data")
print(deliveryData)
x = deliveryData[:, :-1] # -1代表最后一列,这里取到倒数第二列
y = deliveryData[:, -1] # 所有行,和最后一列
print(x)
print(y)
lr = linear_model.LinearRegression() # 线性回归
lr.fit(x, y)
print(lr)
print("coefficients:") # b1 b2
print(lr.coef_)
print("intercept:") # b0
print(lr.intercept_)
xPredict = [102, 6].reshape(1, -1)
yPredict = lr.predict(xPredict)
print("predict:")
print (yPredict)
# predictedY = lr.predict(xPredict)
#
# print("predictedY: " + str(predictedY))
加上车型:转化为001
from numpy import genfromtxt
from sklearn import linear_model
# dataPath=r"Delivery_Dummy.csv"
# data = genfromtxt(datapath,delimiter=",")
dataPath = r"Delivery_Dummy.csv"
deliveryData = genfromtxt(dataPath, delimiter=',') # 转化数据格式
print("data")
print(deliveryData)
x = deliveryData[1:,:-1]
y = deliveryData[1:,-1]
print (x)
print (y)
mlr = linear_model.LinearRegression()
mlr.fit(x, y)
print (mlr)
print ("coef:")
print (mlr.coef_)
print ("intercept")
print (mlr.intercept_)
xPredict = [90,2,0,0,1].reshape(1, -1)
yPredict = mlr.predict(xPredict)
print ("predict:")
print (yPredict)