Python_多元线性回归

import pandas as pd
import numpy as np
from sklearn import linear_model

data = pd.DataFrame({'y':[430,335,520,490,470,210,195,270,400,480],\
                     'x':[30,21,35,42,37,20,8,17,35,25],\
                     'z':[12,10,22,6,8,2,9,8,6,17]})
model = linear_model.LinearRegression()
model.fit(np.array(data[['x','z']]).reshape(-2,2), np.array(data['y']).reshape(-1,1))
x = np.array([35,8]).reshape(-1,2)
a = model.predict(x) #预测