matplotlib之pyplot模块——绘制茎叶图（杆图）stem()

最新推荐文章于 2024-11-25 21:58:26 发布

原创最新推荐文章于 2024-11-25 21:58:26 发布 · 3.9w 阅读

61 ·

CC 4.0 BY-SA版权

文章标签：

#matplotlib #pyplot #stem #茎叶图 #火柴棒图

Matplotlib 专栏收录该内容

124 篇文章

订阅专栏

本文详细介绍了matplotlib库中stem函数的用途、参数及返回值，并通过示例展示了如何绘制不同样式的茎叶图，包括垂直和水平方向的茎叶图，以及自定义茎线、茎头和基线的样式。同时，还展示了如何获取stem函数的返回值进行进一步操作。

当前有效matplotlib版本为：3.4.1。

概述

stem()函数的作用是绘制茎叶图（杆图、棉棒图、火柴杆图）。

茎叶图根据基线（baseline）上的位置（locs）绘制从基线到杆头（heads）的茎线，并在茎头（heads）处放置标记。

函数的签名为matplotlib.pyplot.stem(*args, linefmt=None, markerfmt=None, basefmt=None, bottom=0, label=None, use_line_collection=True, orientation='vertical', data=None)。

函数的调用签名为：stem([locs,] heads, linefmt=None, markerfmt=None, basefmt=None)。

函数的参数为：

locs：对于垂直茎叶图，该参数即茎线的x轴位置，对于水平茎叶图，该参数即茎线的y轴位置。类型为类数组结构，默认值为range(len(heads))。可选参数。
heads：对于垂直茎叶图，该参数即茎头的y值，对于水平茎叶图，该参数即茎头的x值。类型为类数组结构。必备参数。
linefmt：定义茎线的颜色和线形，取值同plot函数。类型为字符串，默认值为'C0-'。可选参数。
markerfmt：定义茎头标记的形状和颜色。类型为字符串，默认值为'C0o'。可选参数。
basefmt：定义基线的格式字符串。类型为字符串，默认值为'C3-'。可选参数。
orientation：茎叶图的方向。类型为字符串，默认值为'vertical'。可选参数。当取值为'vertical'时，茎叶图为垂直方向；当取值为'horizontal'时，则茎叶图为水平方向。
bottom：基线的位置。类型为浮点数，默认值为0。可选参数。
label：图例中使用的标签。类型为字符串，默认值None。可选参数。
use_line_collection：控制线条对象类型，即将被废弃。类型为布尔值，默认值True。可选参数。

返回值为matplotlib.container.StemContainer对象，可以看做markerline, stemlines, baseline三元组。

案例：演示`stem`参数取值

在这里插入图片描述

import matplotlib.pyplot as plt
import numpy as np

# 构造数据
x = np.linspace(0.1, 2 * np.pi, 41)
y = np.exp(np.sin(x))


# 演示默认参数、默认样式
plt.subplot(221)
plt.stem(y)
# 演示茎线、茎头、基线样式
plt.subplot(222)
plt.stem(x, y, linefmt='r--', markerfmt='gD', basefmt='b--', bottom=1)
# 演示水平茎叶图
plt.subplot(223)
plt.stem(x, y, orientation='horizontal')
# 演示stem返回值
plt.subplot(224)
markerline, stemlines, baseline = plt.stem(x, y)
print(markerline.get_color(),  baseline.get_color())
plt.show()