Matplotlib基础图形之散点图

Posted 2020-12-04 keys819

Matplotlib基础图形之散点图

散点图特点：

1.散点图显示两组数据的值，每个点的坐标位置由变量的值决定

2.由一组不连续的点组成，用于观察两种变量的相关性(正相关，负相关，不相关)

3.例如：身高-体重，纬度-温度，等等

示例代码：

import os
import time
import matplotlib.pyplot as plt
basedir = os.path.dirname(os.path.abspath(__file__))
resultdir = os.path.join(basedir, ‘result‘)

class Create_scatter(object):

    def __init__(self,data1,data2,dst,name=‘‘,title=‘Title‘,s=30,c=‘r‘,mark=‘o‘,alpha=1):
        ‘‘‘

        :param data1: x轴使用的数据，一般为列表
        :param data2: y轴使用的数据，一般为列表
        :param dst: 存放生成图片的文件夹路径
        :param name: 图片的名称，默认为当前时间戳
        :param title: 图片的标题，默认为Title
        :param s: 图片的面积，默认为30
        :param c: 图片的颜色，默认为红色
        :param mark: 图片的形状，默认为圆形
        :param alpha: 点的透明度，默认为1，即不透明
        ‘‘‘
        self.x = data1
        self.y = data2
        self.dst = dst
        self.name = name if name else str(int(time.time()))
        self.title = title
        self.s = s
        self.c = c
        self.marker = mark
        self.alpha = alpha

    def drew(self):
        plt.scatter(
            self.x,
            self.y,
            s = self.s,
            c = self.c,
            marker=self.marker,
            alpha = self.alpha
        )
        plt.title(self.title)
        picname = self.name+‘.jpg‘
        picdir = os.path.join(self.dst, picname)
        plt.savefig(picdir)

现在调用上面的类：

height = [155, 160, 165, 170, 175, 180]
weight = [45, 51, 58, 60, 65, 80]
PIC = Create_scatter(height,weight,resultdir,name=‘mypic1‘,c=‘b‘,mark=‘>‘,alpha=0.5)
PIC.drew()

生成图片如下：

 

 关于图片marker，颜色等属性的详细内容，后续会更新