Python 代码:使用 matplotlib 绘制房价与周边学校个数的散点图
这段代码定义了一个名为 'drawLogistic' 的函数,它用于绘制三种城市(北京、上海、深圳)的房价与周边学校个数的散点图。函数接受一个 'self' 参数,表示该函数是类的一个方法。
函数中定义了一个字典 'tarrm',用于保存三种城市的房价和周边学校个数的坐标信息。然后遍历 'self.X_tarr',并将坐标信息存入 'tarrm' 中。最后使用 'matplotlib' 库绘制散点图,并加上相应的标签和图例。
具体代码如下:
def drawLogistic(self):
tarrm = {
0.0: {'x': [], 'y': []},
1.0: {'x': [], 'y': []},
2.0: {'x': [], 'y': []}
}
idx = 0
for tarr in self.X_tarr:
if(tarr[0]>400):
continue
tarrm[self.X_pred[idx]]['x'].append(tarr[0])
tarrm[self.X_pred[idx]]['y'].append(tarr[1])
idx += 1
plt.scatter(tarrm[0.0]['x'], tarrm[0.0]['y'], s=4, c='#130ee7', label='北京', marker='o')
plt.scatter(tarrm[1.0]['x'], tarrm[1.0]['y'], s=8, c='#e605ff', label='上海', marker='d')
plt.scatter(tarrm[2.0]['x'], tarrm[2.0]['y'], s=4, c='#e99900', label='深圳', marker='^')
plt.xlabel('单价')
plt.ylabel('周边学校个数')
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['figure.figsize'] = [12, 8]
plt.legend()
plt.show()
这段代码通过使用 'matplotlib' 库,将不同城市的房价和周边学校个数数据以散点图的形式展示出来,并根据不同的城市使用不同的颜色和形状进行区分,方便直观地观察不同城市之间的差异。
原文地址: https://www.cveoy.top/t/topic/oIgP 著作权归作者所有。请勿转载和采集!