这段代码定义了一个名为 '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' 库,将不同城市的房价和周边学校个数数据以散点图的形式展示出来,并根据不同的城市使用不同的颜色和形状进行区分,方便直观地观察不同城市之间的差异。

Python 代码:使用 matplotlib 绘制房价与周边学校个数的散点图

原文地址: https://www.cveoy.top/t/topic/oIgP 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录