python绘制矩阵茎叶图
时间: 2024-10-03 22:03:09 浏览: 13
在Python中,我们可以使用matplotlib库来绘制矩阵形式的茎叶图,这是一种常见的数据可视化工具,用于展示数值变量的分布。茎叶图将较大的数字分成“茎”部分(通常是十位和个位),较小的数字作为“叶”部分显示。这里是一个简单的例子:
```python
import matplotlib.pyplot as plt
import numpy as np
# 创建一个二维数组作为矩阵数据
data = np.random.rand(5, 5)
# 绘制茎叶图
def stem_graph(data):
for i in range(data.shape[0]):
plt.stem(range(1, data[i].size + 1), data[i], use_line_collection=True)
plt.gca().set_xticks(np.arange(1, len(data[i]) + 1))
plt.xlabel('Stem')
plt.ylabel('Leaf')
plt.title(f'Stem-and-leaf plot for row {i+1}')
plt.grid(True)
plt.show()
stem_graph(data)
```
在这个代码中,`numpy.random.rand()`生成了一个5x5的随机数矩阵。然后我们遍历每一行,用`plt.stem()`函数绘制每个元素的茎叶图,并通过`plt.gca().set_xticks()`设置x轴刻度。