from xlrd import open_workbook
from sklearn import linear_model
book = open_workbook('test.xlsx')
sheet = book.sheet_by_name('Sheet1')
nrows = sheet.nrows
ncols = sheet.ncols
print(nrows,ncols)
xs = []
ys = []
for i in range(1,nrows):
data = []
for j in range(ncols-1):
data.append(float(sheet.cell_value(i, j)))
xs.append(data)
ys.append(float(sheet.cell_value(i, j+1)))
clf = linear_model.LinearRegression()
clf.fit(xs, ys)
print(clf.coef_)
print(clf.intercept_)
运行结果:
601 9
[ 186.95485526 107.97504015 6.70021253 -20.01001596 29.1737441 79.46184922 -68.97587615 -163.44427475]
633.4229176935144