matplotlib绘制折线图与散点图

# -*- coding: utf-8 -*-

import numpy as np
import matplotlib.pyplot as plt

#折线图，figure_no表示显示图的顺序
def simple_line_plot(x,y,figure_no):
    plt.figure(figure_no)
    plt.plot(x,y)
    plt.xlabel('x values')
    plt.ylabel('y values')
    plt.title('Simple Line')
    

def simple_dots(x,y,figure_no):

    plt.figure(figure_no)

#设置线的形状为+，颜色为b蓝色

    plt.plot(x,y,'+b')
    plt.xlabel('x values')
    plt.ylabel('y values')
    plt.title('Simple Dots')

#散点图    

def simple_scatter(x,y,figure_no):
    plt.figure(figure_no)
    plt.scatter(x,y)
    plt.xlabel('x values')
    plt.ylabel('y values')
    plt.title('Simple scatter')

    

def scatter_with_color(x,y,labels,figure_no):

    plt.figure(figure_no)

#设置不同颜色

    plt.scatter(x,y,c=labels)
    plt.xlabel('x values')
    plt.ylabel('y values')
    plt.title('Simple with color')
    

if __name__=="__main__":

#清除之前的图片

    plt.close('all')
    x=np.arange(1,100,dtype=float)
    #y=np.array([np.power(xx,2) for xx in x])
    y=np.sqrt(x)
    
    figure_no=1
    simple_line_plot(x,y,figure_no)
    
    figure_no+=1
    simple_dots(x,y,figure_no)
    
    x=np.random.uniform(size=100)
    #y=np.random.uniform(size=100)
    y=x+1
    
    x1=np.random.uniform(size=100)
    #y=np.random.uniform(size=100)
    y1=x+100
    figure_no+=1
    simple_scatter(x,y,figure_no)
    simple_scatter(x1,y1,figure_no)
    
    

    figure_no+=1

#参生100个随机的0或1，设置颜色

    label=np.random.randint(2,size=100)
    scatter_with_color(x,y,label,figure_no)
    
    plt.show()