python转行系列3:使用matplotlib绘制散点图

前言

         本想练习《机器学习实践》中的相关算法,无奈人老了,算法分析不来了,相关numpy、matplotlib模块的函数也不熟悉,因而先从基本的模块函数开始学习。本人讲述如何使用matplotlib描绘散点图:

cos函数在一个周期(0~2PI)内的散点图。

源码

import matplotlib.pyplot as plt
import numpy as np

def PlotScatterTest1():
    #返回0~2 * np.pi的序列,步长为0.1
    t1 = np.arange(0.0, 2 * np.pi, 0.1)
    #print(t1)

    l = plt.plot(t1, np.cos(t1), 'r_')
    #print(type(l))
    plt.setp(l, markersize=1.0)

    plt.show()

if __name__ == "__main__":
    PlotScatterTest1()

代码解释

  • 程序导入了numpy和matplotlib两个模块(库);
  • 使用numpy模块的arange生成了一个多维(1*N))数组对象(ndarray)
  • 函数plot用于描绘x-y数值对图像,返回一个可绘制的的list 对象。官方解释为:

Plot y versus x as lines and/or markers.

return:

A list of `.Line2D` objects representing the plotted data.

其中,第三个参数’ro’中的’r’代表颜色为红色,’o’表示“圆圈绘制—marker为圆圈”,以下是可选的其他颜色列表。具体请查看plot的帮助信息

    ``'b'``          blue
    ``'g'``          green
    ``'r'``          red
    ``'c'``          cyan
    ``'m'``          magenta
    ``'y'``          yellow
    ``'k'``          black
    ``'w'``          white
  • plt.setp函数表示设置刚才绘图的属性,设置maker大小为1
  • plt.show()表示图像

显示结果

python转行系列3:使用matplotlib绘制散点图

将plot函数的第三个参数'ro'修改为'r_' 红色水平线绘制后显示如下:

python转行系列3:使用matplotlib绘制散点图

总结

         plot、setp函数还有很多参数,具体请参阅其帮助信息。