python波士顿房价热力图
时间: 2023-11-04 12:01:17 浏览: 137
基于Python的热力图生成
python波士顿房价热力图是通过使用pyecharts库中的HeatMap模块来创建的。首先,导入所需的库和模块,并读取波士顿房价数据文件。然后,使用pandas的corr()方法计算出每列之间的相关系数,并将相关系数数据转换为列表形式。接下来,使用HeatMap模块的相关方法将数据添加到热力图中,并设置相关的图表选项。最后,使用render_notebook()方法将热力图渲染出来。您可以根据需要自定义标题、颜色映射等参数。以下是一个示例代码:
import pandas as pd
from pyecharts import options as opts
from pyecharts.charts import HeatMap
data1 = pd.read_csv('boston_house_prices.csv', encoding='gbk')
df_corr = data1.corr()
data = df_corr.values.tolist()
x_index = df_corr.index.tolist()
y_index = df_corr.columns.tolist()
value = [[i, j, round(data[i][j], 2)] for i in range(len(x_index)) for j in range(len(y_index))]
heatmap = HeatMap(init_opts=opts.InitOpts())
heatmap.add_xaxis(y_index)
heatmap.add_yaxis("", x_index, value, label_opts=opts.LabelOpts(is_show=True, position="inside"))
heatmap.set_global_opts(
title_opts=opts.TitleOpts(title="波士顿房价热力图"),
visualmap_opts=opts.VisualMapOpts(max_=1),
xaxis_opts=opts.AxisOpts(axislabel_opts=opts.LabelOpts(rotate=30))
)
heatmap.render_notebook()
阅读全文