python預測房價

numpy教程:

http://www.jb51.net/article/49397.htm

線性迴歸教程:http://wiki.mbalib.com/wiki/線性迴歸預測法

 # Enter your code here. Read input from STDIN. Print output to STDOUT
#coding:utf-8
import numpy
import string
from numpy import *
f=open("/dev/stdin",'r')
j=f.readline()
N=int(j[2])
F=int(j[0])
a=[]
for i in range(N):
 b=f.readline()
 c=b.split()
 for item in c:
  d=string.atof(item)
  a.append(d)

#a是一個輸入序列
#將輸入的序列變爲矩陣,然後變爲數組

in_array=numpy.array(a).reshape(N,F+1)

#開始用普通最小二乘法計算參數
X=in_array[:,:F]
Y=in_array[:,F]
x=mat(X)
y=mat(Y)
parameter=(x.T*x).I*x.T*(y.T)
 

#====================================================
#獲取待估測值
a1=[]
m=f.readline()

T=int(m)  
for i in range(N,N+T):
 b1=f.readline()
 c1=b1.split()
 for item in c1:
  d1=string.atof(item)
  a1.append(d1)

o_array=numpy.array(a1).reshape(T,F)

house_price=mat(o_array)*parameter
for i in range(T):
    print house_price[i,0]
       


發佈了58 篇原創文章 · 獲贊 6 · 訪問量 9萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章