爬取链家网房源信息线性回归做房价预测（二）

最新推荐文章于 2021-10-26 13:40:11 发布

原创

最新推荐文章于 2021-10-26 13:40:11 发布 · 1.7k 阅读

26 ·

CC 4.0 BY-SA版权

文章标签：

#lasso回归 #爬虫

1 爬取链家网房源数据

https://blog.youkuaiyun.com/weixin_41044499/article/details/94382662

得到的数据格式如下：

2 将所有的特征进行处理，转化为多个特征。调用skilearn的linear-regression方法，做房价的线性回归的训练和预测，

# !/usr/bin/python
# -*- coding:utf-8 -*-

from sklearn.linear_model import LinearRegression
import pandas  as pd
import numpy as np
raw = pd.read_excel('lianjia-info.xlsx')
#房价 户型 平米 朝向 装修 电梯
raw['装修2'] = raw['装修'].map({'精装':3,'简装':2,'毛坯':1,'其他':0})
#将不同的朝向分别单独出来一列
raw['nan']= raw['朝向'].apply(lambda x:'南' in x).map({True:1,False:0})
raw['dong']= raw['朝向'].apply(lambda x:'东' in x).map({True:1,False:0})
raw['xi']= raw['朝向'].apply(lambda x:'西' in x).map({True:1,False:0})
raw['bei']= raw['朝向'].apply(lambda x:'北' in x).map({True:1,False:0})
#print(raw['bei'