matplotlib可视化篇stem()--棉棒图

棉棒图主要用来显示一个包含正负的数据集，如在地震勘探中表示地层间反射系数，官方项目地址，官方示例

函数：matplotlib.pyplot.stem(x,y, linefmt=None, markerfmt=None, basefmt=None)
主要参数：
x: 棉棒的x轴基线的取值范围
y: 棉棒的长度
linefmt: 棉棒的样式，可选择{'-','--',':','-.'}，根据实际需求来选择
markerfmt: 棉棒末端的样式
basefmt: 指定基线的样式
label: 图例显示内容
示例：

import numpy as np
import matplotlib.pyplot as plt
# 生成模拟数据集
x=np.linspace(0,10,20)
y=np.random.randn(20)
# 绘制棉棒图
markerline, stemlines, baseline = plt.stem(x,y,linefmt='-',markerfmt='o',basefmt='--',label='TestStem')
# 可单独设置棉棒末端，棉棒连线以及基线的属性
plt.setp(markerline, color='k')#将棉棒末端设置为黑色

plt.legend()
plt.show()

效果图：

棉棒图

针对棉棒图的显示效果可以自己添加特定属性的设置，如设置stemlines参数即设置棉棒图连接线，而设置baseline参数会修改基线属性。