matplotlib的pyplot的使用

pyplot几个常用函数介绍

• figure 创建一个图，用于标识下面的绘制在这个图中
• axis 描述 横坐标和纵坐标的显示区间， 举例： [-20, 20, -100, 600] , 横坐标在[-20， 20]，纵坐标在[-100, 600]
• plot 将数据写到该图中，plot的前两个参数分别是横坐标数组，纵坐标数组。 举例plot(x, y)，将x序列，y序列 以(x, y) 数值对 ，绘制到该图中
• show 显示该图

下面给出一个实例

def test_plot():
    plt.figure()
    draw_axis(plt, [-20, 20], [-100, 100])
    x = [] # 储存横坐标值
    y = [] # 储存纵坐标值
    for key in range(-20, 20):
        if key <= 0:
            value = key*2 + 1 # 即 y = 2x + 1的函数
        else:
            value = 0.25*key*key + 1  # 即 y = 0.25*x^2 + 1的函数
        # 分别保存横纵坐标
        x.append(key)
        y.append(value)
    plt.axis([-20, 20, -100, 100])
    plt.plot(x, y)
    plt.show()

运行的效果图如下：

实例图

pyplot更多函数介绍

• scatter 这个函数是绘制散点
  前两个参数是输入x, y的坐标序列，后面s=定义点的大小，c=定义颜色。余下不赘述，用到再研究。
  见下面的示例：

xx = [40920.0, 14488.0, 26052.0, 75136.0, 38344.0, 72993.0, 35948.0, 42666.0, 67497.0, ..., 43757.0]
yy = [8.326976, 7.153469, 1.441871, 13.147394, 1.669788, 10.14174, 6.830792, 13.276369, ..., 7.882601]

def test_plot():
    plt.scatter(xx, yy, s=100, c='r') # 点的大小设为100， 颜色为红色
    plt.show()

效果图如下：

效果图

现在将大小、颜色作为参数传入，

weight = [3, 2, 1, 1, 1, 1, 3, 3, 1, ..., 3]
def test_plot():
    # s和c都支持array和list的类型参数，但是array类型的更好用，因为支持给weight统一乘以15，将点的大小区分开。至于颜色，貌似只要数值不同，会默认分配几种不同颜色
    plt.scatter(xx, yy, s=15.0*array(weight), c=weight)
    plt.show()

效果图如下：

添加大小和颜色