import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn import linear_model
# 加载数据集
data = pd.read_csv('C:\\Users\\xiongwei\\Desktop\\汽车销售数据.csv')
data = data.dropna()
y = data.iloc[:,2]
x = data.iloc[:, 3:]
# 拆分训练数据集和测试数据集
X_train, X_test, y_train, y_test = train_test_split(x, y,test_size = 0.25, random_state = 1)
regr = linear_model.LinearRegression()
regr.fit(X_train,y_train)
print('coefficients(b1,b2...):',regr.coef_)
print('intercept(b0):',regr.intercept_)
# 预测数据
y_pred = regr.predict(X_test)
# 查看模型得分
print(regr.score(X_test, y_test))
输出