Lesson 16: Multiple Regression with Python

1063593118

======================

Code

============================

#! usr/bin/env python2.7

# ———–Read x and y————
x1=[]
x2=[]
x3=[]
y=[]

import csv
with open(‘price.csv’) as csvfile:
reader=csv.reader(csvfile)
for row in reader:
xy=row
x1.append(int(xy[0]))
x2.append(int(xy[1]))
x3.append(int(xy[2]))
y.append(int(xy[3]))
x=[x1,x2,x3]

print ‘=======================================’
print ‘Define X1=Bedrooms X2=Bathrooms X3=SqFt’
print ‘=======================================’
print

#————-Calculate Multiple Regression Result————
import numpy as np
import statsmodels.api as sm

def reg_m(y, x):
x=np.array(x).T
X=sm.add_constant(x)
results=sm.OLS(endog=y,exog=X).fit()
return results

print reg_m(y,x).summary()

留下评论