前言
本文档旨在记录通过python的matplotlib包绘制散点图的实现细节
需求分析
1.将excel的数据导入python
2.通过python绘制散点图
1)不同类别的点标注不同的颜色;
2)绘制图例;
具体实现
需求1:将excel的数据导入python
然而这一次的excel文件中数据长度不一致,python会读取长度为最长列长度(MaxRow)的所有数据,如果某些列的长度小于最长列长度(MaxRow),python会将其处理为空值;空值为str,这样运行会报错说str和float类型不一致,如果直接处理成0会有问题;
因此,我们需要去除空值,通过filter(None,(你的List)来去除空值和None;
需求2:绘制散点图
散点图在之前的博文里也实现了,具体参见同上链接;
今天的散点图要给不同的类别添加上不同的颜色,实际上实现很简单,直接多些几行plt.scatter即可。
需求3:绘制图例
通过plt.legend()实现,要想写明图例当中的标注,需要在plt.scatter(label='')写明label的参数即可。
可以通过设置loc的参数来改变图例的位置,具体如下: