图表可视化

stem （杆图、棉棒图、火柴杆图）

相关文档:

• setp描述

• stem描述

    import matplotlib.pyplot as plt
    import numpy as np
    
    # time domain in which we sample
    x = np.linspace(0, 20, 50)
    # random function to simulate sampled signal
    y = np.sin(x + 1) + np.cos(x ** 2)
    
    x1 = np.linspace(5, 15, 50)
    # random function to simulate sampled signal
    y1 = np.sin(x1) + np.cos(x1 ** 1)
    
    # here we can setup baseline position
    bottom = -0.1
    
    # set label for legend.  stem（杆图、棉棒图、火柴杆图）
    # 绘制两张stem 2D图表
    markerline, stemlines, baseline = plt.stem(x, y, bottom=bottom,label='A')
    markerline2, stemlines2, baseline2 = plt.stem(x1, y1, bottom=bottom,label='B')
    
    # we use setp() here to setup
    # multiple properties of lines generated by stem()
    # setp相当于给曲线更改属性，比如颜色，大小之类的
    plt.setp(markerline, color='red', marker='o')
    plt.setp(stemlines, color='blue')
    plt.setp(baseline, color='grey', linewidth=2)
    
    plt.setp(markerline2, color='black', marker='o')
    plt.setp(stemlines2, color='green')
    plt.setp(baseline2, color='grey', linewidth=2)
    # draw a legend
    plt.legend()
    plt.show()