引言

随着城市化进程的加速，人口流动日益频繁，租房市场作为城市生活的重要组成部分，其价格波动对居民生活质量和城市经济发展具有显著影响。成都市，作为中国西部地区的经济、文化、交通和科技中心，近年来吸引了大量人才和企业，租房需求持续增长。然而，租房价格的不确定性给租户和房东带来了诸多不便。为了更好地理解租房市场的动态，预测租房价格成为一项重要的研究课题。

研究方法

本文将采用多种数据分析方法，包括统计分析、机器学习等，以确保模型的准确性和可靠性。

研究目标

本文旨在通过构建基于成都市各区（市）县的租房价格预测模型，分析影响租房价格的关键因素。

数据信息

数据源自某租房平台网站，数据为公开数据，通过整合形成表格，本数据需要进行清洗，没有国外数据好用。

过程……

读入数据：

import matplotlib.pylab as plt
import re
import numpy as np
import seaborn as sns
import pandas as pd
plt.rcParams['font.sans-serif'] = ['SimHei'] #解决中文显示
plt.rcParams['axes.unicode_minus'] = False   #解决符号无法显示
import warnings
warnings.filterwarnings('ignore')
# 导入plotly库，这是一个基于Web的交互式图表库，允许创建丰富的、交互式的数据可视化图表。  
import plotly as py    
# 导入cufflinks库，这是一个用于Pandas DataFrame的Plotly绘图接口，可以让Pandas DataFrame直接通过cufflinks的API绘制Plotly图表。  
import plotly.express as px
import plotly.graph_objects as go  


df = pd.read_csv('成都.csv')
df.tail()

原始数据长下面这样，显示后5行：

我们先看下数据信息、重复值、缺失值情况：
有4列存在缺失值情况，181行数据重复。

# 删除缺失值
df.dropna(inplace=True)
# 删除重复值
df.drop_duplicates(inplace=True)
df.shape

删完剩余2786行, 8列。

数据清洗：

# 定义一个函数，用于从字符串中提取括号前的中文部分  
def extract_chinese_before_brackets(s):  
    # 检查传入的参数s是否为字符串类型  
    if isinstance(s, str):  
        # 使用正则表达式搜索字符串s，查找从开头到第一个全角左括号（之前的所有内容  
        match = re.search(r'^(.*?)（', s)  
        # 如果找到了匹配项  
        if match:  
            # 返回匹配到的内容（即括号前的所有字符）  
            return match.group(1)  
        else:  
            # 如果没有找到匹配项（即没有括号或括号前没有内容），返回空字符串  
            return ''  
    else:  
        # 如果传入的参数s不是字符串类型，直接返回空字符串  
        return ''  
  
# 应用上面定义的函数到DataFrame的'楼层'列  
# 使用apply方法，将函数应用于'楼层'列的每一个元素  
# 并将函数的返回值（即每个元素括号前的中文部分）存储在新的列'楼层类型'中  
df['楼层类型'] = df['楼层'].apply(extract_chinese_before_brackets)  

df.tail()

上面定义一个函数，用于提取“楼层”列最左边的中文文字，存储于新建列“楼层类型”列并显示后5行，详见下图：

# 使用str.extract方法从'楼层'列的每个字符串中提取括号内的数字（\d+）  
# 这里的正则表达式'（(\d+)层）'用于匹配全角左括号'（'后跟着一个或多个数字'\d+'，然后是全角右括号和'层'字  
# 提取的数字（即括号内的内容）将被存储在新的列'楼房层数'中  
df['楼房层数'] = df['楼层'].str.extract(r'（(\d+)层）')    
# 使用drop方法删除原始的'楼层'列  
# 参数columns='楼层'指定了要删除的列名  
# inplace=True表示在原地修改DataFrame，而不是返回一个新的DataFrame  
df.drop(columns='楼层', inplace=True)    
# 使用head方法查看修改后的DataFrame的前几行（默认是前5行）  
# 这有助于验证'楼房层数'列是否已正确添加，并且'楼层'列是否已被删除  
df.head()

再利用正则表达式，提取出括号内的数字，也就是楼层并存储于新列，显示前5行数据，详见下图：

# 提取的结果将被分配到新的列'卧室间数'、'客厅间数'和'卫生间数'
df[['卧室间数', '客厅间数', '卫生间数']] = df['户型'].str.extract(r'(\d+)室(\d+)厅(\d+)卫', expand=True)  
df.drop(columns='户型', inplace=True)
df.head()

新建三列’卧室间数’, ‘客厅间数’, '卫生间数’用于存储户型内的如“3室1厅2卫 ”便于后续的机器学习，还是利用正则表达式提取，详见下图：

df[['区（市）县', '地域', '楼盘名称']] = df['位置'].str.split('-', expand=True, n=2)
df.drop(columns='位置', inplace=True)
df.head()

新建’区（市）县’, ‘地域’, '楼盘名称’三列，用于存储“位置”列如“青羊-外光华-凯德风尚 ”用于后续的数据分析，详见下图：

现在我们来看看“朝向”列有哪些不重复的唯一值：
可以看到国内的数据很乱，现在要做的是有“东”或“西”字眼的我们统一把他替换为“东西”，有“南”或“北”字眼的我们统一把他替换为“南北”。继续写个函数来处理数据。

# 定义转换函数  
def transform_direction(direction):  
    if pd.isna(direction):  # 检查是否为 NaN  
        return np.nan  # 或者你可以选择返回其他默认值，如 '未知'  
    if isinstance(direction, str):  # 确保是字符串类型  
        if '东' in direction or '西' in direction:  
            return '东西'  
        elif '南' in direction or '北' in direction:  
            return '南北'  
        else:  
            return direction  # 返回原始字符串（这里可能包含'未知'等）  
    else:  
        return np.nan  # 如果不是字符串类型，返回 NaN（或其他默认值）  
  
# 将 '朝向' 列转换为字符串类型，并处理 NaN 值  
df['朝向'] = df['朝向'].astype(str)    
# 应用转换函数到 Series 的每个元素  
df['朝向'] = df['朝向'].apply(transform_direction)  
df.tail()

处理完成，显示后5行，详见下图：

再看下其他列有没有问题：
没啥问题，现在将数字类型的列转为相对应的类型：

df = df.astype({
          '金额': 'int',
          '楼房层数': 'int',
          '卧室间数': 'int',
          '客厅间数': 'int',
          '卫生间数': 'int'
          }) # 指定字段转指定类型
df.info()

搞定，进入下一阶段内容。

探索性分析（EDA）

# 使用Plotly绘制箱型图  
fig = go.Figure(data=[go.Box(  
    y=df['金额'],  # 这里y轴对应你的租金数据  
    name='房屋租金'    # 图例名称  
)])  
  
# 自定义图表标题和轴标签  
fig.update_layout(title='房屋租金——箱型图',  
                   yaxis_title='租金',  
                   xaxis_title='')  # 箱型图通常不需要x轴标题，除非你有多个箱型图并列  
  
# 显示图表  
fig.show()

从上图可以看出，大部分房屋月租金都比较集中，成都市最高的租金为45000元/月，最小的为500元/月，中位数为2500元/月，还有四分位距、决策边界等内容。

# 对DataFrame按'区（市）县'进行分组，并获取每个组的'房屋租金'数据  
grouped = df.groupby('区（市）县')['金额']  
  
# 创建一个空的图表对象  
fig = go.Figure()  
  
# 遍历每个组，为每个组添加一个箱型图轨迹  
for name, group in grouped:  
    fig.add_trace(go.Box(  
        y=group,  
        name=name,  
        boxpoints='all',  # 显示所有点  
        jitter=0.3,       # 点的抖动量，以防重叠  
        pointpos=-1.8      # 点的位置  
    ))  
  
# 自定义图表布局  
fig.update_layout(  
    title='按区（市）县分组的房屋租金——箱型图',  
    yaxis_title='租金',  
    xaxis_title='',  
    xaxis=dict(  
        type='category',  # 设置x轴为类别轴  
        categoryorder='array',  # 自定义类别顺序  
        categoryarray=list(grouped.groups.keys())  # 将dict_keys转换为列表  
    )  
)  
  
# 显示图表  
fig.show()

分区县绘制的箱型图，我们重点关注一下“三遗”之城都江堰，最高的房租为6000元/月。

from scipy.stats import gaussian_kde  
  
# 假设df是你的DataFrame，且包含'房屋面积'列  
  
# 提取房屋面积数据  
x = df['房子大小'].values  
  
# 计算KDE  
kde = gaussian_kde(x)  
# 生成KDE曲线的x轴数据点（这里假设你想要覆盖的范围是x的最小值到最大值，步长为0.1）  
x_kde = np.linspace(x.min(), x.max(), 300)  
# 计算KDE曲线在这些点上的y值  
y_kde = kde(x_kde)  
  
# 绘制直方图  
fig = go.Figure()  
fig.add_trace(go.Histogram(  
    x=x,  
    histnorm='probability density',  # 使得直方图的面积等于1，与KDE曲线可比较  
    name='直方图'  
))  
  
# 绘制KDE曲线  
fig.add_trace(go.Scatter(  
    x=x_kde,  
    y=y_kde,  
    mode='lines',  
    name='KDE曲线'  
))  
  
# 自定义图表布局  
fig.update_layout(  
    title='房屋面积分布（直方图+KDE）',  
    xaxis_title='房屋面积',  
    yaxis_title='概率密度'  
)  
  
# 显示图表  
fig.show()

要解读这种图表，需要观察直方图中条形的高度和KDE曲线的形状。如果直方图的条形较高，表示在相应的面积区间内房屋较多；KDE曲线的峰值则表示最可能的房屋面积。通过比较直方图和KDE曲线，可以分析房屋面积的分布模式，例如是否存在一个或多个常见的面积区间，或者面积分布是否均匀。

# 创建一个散点图  
fig = go.Figure(data=go.Scatter(  
    x=df['房子大小'],  
    y=df['金额'],  
    mode='markers',  # 设置为散点图  
    marker=dict(  
        size=10,     # 散点大小  
        color='blue',# 散点颜色  
        opacity=0.8  # 散点透明度  
    )  
))  
  
# 自定义图表布局  
fig.update_layout(  
    title='房屋面积与租金——散点图',  
    xaxis_title='房屋面积',  
    yaxis_title='房屋租金'  
)  
  
# 显示图表  
fig.show()

散点图主要是解释相关性的图表，不懂的可以点击链接转到我的博客详细查看。蓝色框框内左边的数字是面积，右边是房租。
上图实际就是一个坐标轴，分x和y轴，每个点代表每个数值，单个数据叫标量，一组数据叫向量，坐标轴内可以视作向量空间。

# 计算每个出租方式的计数  
value_counts = df['类型'].value_counts()  
  
# 创建饼图数据  
labels = value_counts.index.tolist()  # 获取标签（出租方式）  
values = value_counts.values.tolist()  # 获取计数  
  
# 绘制饼图  
fig = go.Figure(data=[go.Pie(labels=labels, values=values, hole=0.3, textinfo='label+percent', insidetextorientation='radial')])  
  
# 自定义图表布局  
fig.update_layout(  
    title='出租方式分布——饼图',  
    font_size=12,  # 你可以根据需要调整字体大小  
    legend_title_text='出租方式'  # 如果你需要图例（对于饼图通常不需要，但这里作为示例）  
)  
  
# 注意：饼图通常不需要图例，因为每个切片都直接标记了标签和百分比  
# 如果你确实需要图例（尽管这在饼图中不常见），你可能需要以一种不同的方式来实现它  
  
# 显示图表  
fig.show()

从上图可以看出，整租占比达到80.7%。

# 创建箱形图数据  
fig = go.Figure()  
  
# 遍历'出租方式'的唯一值，为每个类别绘制一个箱形图  
for category in df['类型'].unique():  
    subset = df[df['类型'] == category]  
    fig.add_trace(go.Box(  
        y=subset['金额'],  
        name=category,  
        marker=dict(  
            color='rgb(0, 0, 0)',  # 箱体的颜色  
            outliercolor='rgba(219, 64, 82, 0.6)',  # 异常值的颜色  
            line=dict(  
                color='rgb(0,0,0)',  
                width=1.5  
            )  
        ),  
        boxmean='sd'  # 'sd'表示使用标准差标记平均值，你也可以设置为'mean'来直接使用平均值  
    ))  
  
# 自定义图表布局  
fig.update_layout(  
    title='按出租方式分组的房屋租金——箱形图',  
    yaxis_title='房屋租金',  
    xaxis_title='出租方式',  
    boxmode='group'  # 箱形图的显示模式，'group'表示每个类别的箱形图并排显示  
)  
  
# 显示图表  
fig.show()

从上图可以看出，整租的租金浮动较大，而合租的租金区间很小。合租的租金价格在2100元/月以下区间，仅有1个房屋租金是2100元/月，其余都往下在走。

sns.barplot(data=df,x='朝向',y='金额')
plt.show()

南北朝向的房子明显租金更高。

# 计算每个楼层的计数  
floor_counts = df['楼层类型'].value_counts()  
  
# 创建饼图数据  
fig = go.Figure(data=[go.Pie(  
    labels=floor_counts.index,  # 标签是楼层的唯一值  
    values=floor_counts.values,  # 值是每个楼层的计数  
    hole=0.3,  # 设置饼图的中心空洞大小  
    textinfo='label+percent',  # 在饼图上显示标签和百分比  
    insidetextorientation='radial'  # 文本方向为径向  
)])  
  
# 更新图表的布局  
fig.update_layout(  
    title='楼层分布——饼图',  # 设置图表的标题  
    annotations=[dict(text=str(round(value, 2))+'%', x=0.5, y=0.5,  
                      xref='paper', yref='paper',  
                      showarrow=False, font=dict(size=10),  
                      textangle=0 if i == 0 else i*360/len(floor_counts))  
                 for i, value in enumerate(floor_counts.values)],  
    # 注释文本的位置需要手动调整，这里使用了一种简单的方法，但可能需要根据实际情况调整  
)  
  
# 显示图表  
fig.show()  

上图展示了成都市楼层分布情况。

sns.barplot(data=df,x='楼层类型',y='金额')
plt.show()

低楼层租金略微高于中楼层，中楼层租金略微高于高楼层。

# 相关性分析
numeric_columns_simplified = df.select_dtypes(include='number')
sns.heatmap(numeric_columns_simplified.corr(),vmax=1,annot=True,linewidths=0.5,cbar=False,cmap='YlGnBu',annot_kws={'fontsize':12})
plt.xticks(fontsize=10)
plt.yticks(fontsize=10)
plt.title('房屋特征皮尔逊相关系数矩阵——热力图',fontsize=17)
plt.show()

跟房租最相关的是他自己，1也就是100%相关，第二相关的是房子大小68%，需要注意的是楼房层数是负相关，也就是楼层越高房屋租金越低。

from wordcloud import WordCloud  
# 将'楼层信息'列中的所有文本合并成一个长字符串  
text = ' '.join(df['楼盘名称'].astype(str))  
  
font_path = '/Library/Fonts/Arial Unicode.ttf'  # macOS上的示例路径  
 
  
# 创建词云对象，并指定中文字体  
wordcloud = WordCloud(width=800, height=400, background_color='black', font_path=font_path).generate(text)  
  
# 使用matplotlib显示词云  
plt.figure(figsize=(10, 5))  
plt.imshow(wordcloud, interpolation='bilinear')  
plt.axis('off')  # 不显示坐标轴  
plt.show()

根据楼盘名称绘制的词云图，可以看出，“保利天悦”出现的频率很高。

机器学习

特征工程：

# 特征筛选
new_df = df[['金额', '区（市）县', '类型', '朝向', '楼层类型', '房子大小',  '楼房层数', '卧室间数', '客厅间数', '卫生间数']]
new_df = new_df.astype('str')
new_df

选取以下内容作为特征：

很多中文，我们需要把他转为数字，变成计算机熟悉的分类数据：

from sklearn.preprocessing import LabelEncoder  
  
# 定义一个函数，用于对pandas Series进行标签编码  
def label_encode(series):  
    # 创建一个LabelEncoder实例  
    le = LabelEncoder()  
    # 使用LabelEncoder的fit_transform方法对Series进行拟合和转换  
    # 这将Series中的类别（通常是字符串）转换为从0到n_classes-1的整数  
    transformed_series = le.fit_transform(series)  
    # 返回转换后的Series  
    return transformed_series  
  
# 假设new_df是一个已经存在的DataFrame  
# 使用apply函数将label_encode函数应用到DataFrame的每一列上  
# axis=0表示函数沿着列（垂直方向）应用，即每一列被视为一个Series进行处理  
# 注意：这里直接使用apply可能会导致一些问题，特别是如果DataFrame包含非对象类型的列  
# 因为LabelEncoder只能用于对象类型（如字符串）的列  
# 但为了注释的目的，我们假设所有列都是对象类型  
new_df = new_df.apply(label_encode, axis=0)
new_df

上图已经全部为了数字。

划分训练集和测试集：

from sklearn.model_selection import train_test_split
# 准备数据
X = new_df.drop('金额',axis=1)
y = new_df['金额']
# 划分数据集
X_train,X_test,y_train,y_test = train_test_split(X,y,test_size=0.2,random_state=42)
print('训练集大小:',X_train.shape[0])
print('测试集大小:',X_test.shape[0])

定义模型评估函数：

from sklearn.metrics import r2_score, mean_absolute_error, mean_squared_error  
# 导入sklearn库中用于评估模型性能的三个指标：R^2分数、平均绝对误差(MAE)、均方误差(MSE)  
  
# 定义一个训练模型并输出模型的评估指标的函数  
def train_model(ml_model):  
    # 打印传入的模型类型或实例  
    print("模型名称: ", ml_model)        
    # 使用训练数据拟合模型。这里假设X_train和y_train是全局变量或已在此函数外部定义  
    model = ml_model.fit(X_train, y_train)        
    # 在训练数据上评估模型的性能（这一步主要用于调试或查看过拟合情况，通常不用训练集评估模型最终性能）  
    print("模型预测准确率: ", model.score(X_train, y_train))        
    # 使用训练好的模型对测试集进行预测  
    predictions = model.predict(X_test)        
    # 计算并打印R^2分数，评估模型的预测性能  
    r2score = r2_score(y_test, predictions)  
    print("R方: ", r2score)        
    # 计算并打印平均绝对误差(MAE)，衡量预测值与真实值之间差异的平均绝对值  
    print('平均绝对误差（MAE）:', mean_absolute_error(y_test,predictions))        
    # 计算并打印均方误差(MSE)，衡量预测值与真实值之间差异的平方的平均值  
    print('均方误差（MSE）:', mean_squared_error(y_test,predictions))        
    # 计算并打印均方根误差(RMSE)，是MSE的平方根，与预测值的尺度相同，便于理解  
    print('均方根误差（RMSE）:', np.sqrt(mean_squared_error(y_test,predictions)))  
    # 真实值和预测值的差值(绘制残差分布图)
    sns.distplot(y_test - predictions)

构建线性回归模型：

# 导入LinearRegression类  
from sklearn.linear_model import LinearRegression    
# 创建LinearRegression的实例  
lg = LinearRegression()    
# 调用train_model函数，传入LinearRegression实例作为参数  
train_model(lg)

构建决策树模型：

# 构建决策树回归
from sklearn.tree import DecisionTreeRegressor
# 初始化DecisionTreeRegressor并设置防止过拟合的参数  
tree = DecisionTreeRegressor()  
train_model(tree)

构建xgboost模型：

# 构建xgboost回归模型
from xgboost import XGBRegressor
xgb = XGBRegressor(learning_rate=1.4, n_estimators=1000) 
train_model(xgb)

计算各特征重要程度：

# 首先，定义特征标签列表，这里假设X_train是一个DataFrame，其列名即为特征名  
feat_labels = X_train.columns[0:]  # 获取X_train的所有列名作为特征标签    
# 注意：这里假设xgb是一个已经训练好的XGBoost模型实例，该模型具有feature_importances_属性  
# feature_importances_属性包含了模型中每个特征的重要性评分  
# importances = xgb.feature_importances_  
importances = tree.feature_importances_  
# 使用argsort函数对重要性评分进行排序，并通过[::-1]反转数组，以得到重要性从高到低的索引  
# 这样，indices数组中的第一个索引对应最重要的特征  
indices = np.argsort(importances)[::-1]  
  
# 初始化两个列表，用于存储特征和它们对应的重要性评分  
index_list = []  
value_list = []  
  
# 遍历排序后的索引和特征数量（X_train.shape[1]），将特征和重要性评分添加到列表中  
# 注意：range(X_train.shape[1])实际上生成的是从0到特征数-1的索引，但在这里我们将其与indices结合使用  
# 并在打印时添加了1，以便特征编号从1开始显示（如果偏好从0开始，则可以去掉f+1）  
for f, j in zip(range(X_train.shape[1]), indices):  
    index_list.append(feat_labels[j])  # 将特征名添加到index_list  
    value_list.append(importances[j])  # 将重要性评分添加到value_list  
    print(f + 1, feat_labels[j], importances[j])  # 打印特征编号、名称和重要性评分  
  
# 使用matplotlib的pyplot模块创建一个图形  
plt.figure(figsize=(10,6))  # 设置图形的大小为10x6英寸  
  
# 绘制水平条形图，展示特征的重要性评分  
# 注意：这里使用了[::-1]来反转列表，因为matplotlib默认是从下到上绘制条形图  
# 但我们想要从最重要的特征开始，即从图形顶部开始绘制  
plt.barh(index_list[::-1], value_list[::-1])  
  
# 设置y轴刻度标签的字体大小  
plt.yticks(fontsize=12)  
  
# 设置图形的标题和字体大小  
plt.title('各特征重要程度排序', fontsize=14)  
  
# 显示图形  
plt.show()

从下图可以看出，影响房屋租金特征排序最高的是房子大小，接着是楼房层数，详见下图：

# 模型预测
y_pred = tree.predict(X_test)
result_df = pd.DataFrame()
result_df['真实值'] = y_test
result_df['预测值'] = y_pred
result_df.sample(2)

随机选择两个真实值和预测值对比，对了1个，感觉模型有些过拟合了，想要再准点，还需要对模型调超参数，由于比较复杂，今天就不展示了。

还有一个知识点需要给大家讲，下面其实是一个错误示范，之前中文转数字对数据进行了标准化，但在预测时未对y_pred进行相应的逆变换，预测值看起来比原始数据（房屋租金）小了很多，所以下面的数值是标准化后的数值，并不是原始数据。

数据可视化（真实值和预测值对比）：

# 创建一个新的图形，并设置其大小为10x6英寸  
plt.figure(figsize=(10,6))      
# 绘制预测值。这里只取前200个预测值（y_pred[:200]）和对应的索引（range(len(y_test))[:200]）  
# 使用蓝色('b')线条表示预测值，并设置图例标签为'predict'  
plt.plot(range(len(y_test))[:200], y_pred[:200], 'b', label='预测值')    
# 绘制测试集的真实值。同样只取前200个真实值（y_test[:200]）和对应的索引  
# 使用红色('r')线条表示真实值，并设置图例标签为'test'  
plt.plot(range(len(y_test))[:200], y_test[:200], 'r', label='真实值')    
# 添加图例，并设置其位置在图形的右上角，字体大小为15  
plt.legend(loc='upper right', fontsize=15)    
# 设置x轴的标签为'the number of house'，并通过fontdict字典设置其字体权重和大小  
plt.xlabel('the number of house', fontdict={'weight': 'normal', 'size': 15})    
# 设置y轴的标签为'value of Price'，同样通过fontdict字典设置其字体权重和大小  
plt.ylabel('value of Price', fontdict={'weight': 'normal', 'size': 15})    
# 显示图形  
plt.show()

本文数据集可以点击本文上方下载，创作不易，点赞、评论、收藏是我创作的动力！