棉棒图主要用来显示一个包含正负的数据集,如在地震勘探中表示地层间反射系数,官方项目地址,官方示例
函数: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()
效果图:
棉棒图