Python制作茎叶图

茎叶图（stem-and-leaf plot）是一种用于展示数据分布的可视化方法，它将每个数据点的小数位分成两部分，一部分作为“茎”，一部分作为“叶”，并将它们排列在一起以形成类似于直方图的图形。Python中可以使用numpy和matplotlib库来绘制茎叶图。

以下是一个简单的示例代码：

import numpy as np
import matplotlib.pyplot as plt

# 生成一组随机数据
data = np.random.randint(1, 100, size=20)

# 取出每个数据点的十位和个位
stems = np.floor(data / 10)
leafs = data % 10

# 将数据点按照茎叶分布进行排序
order = np.argsort(stems)
sorted_stems = stems[order]
sorted_leafs = leafs[order]

# 绘制茎叶图
plt.figure()
plt.stem(sorted_stems, sorted_leafs, use_line_collection=True)
plt.xlabel('Stems')
plt.ylabel('Leafs')
plt.title('Stem-and-Leaf Plot')

plt.show()

运行上述代码，将生成一个茎叶图，类似于下图所示：