python绘制堆叠柱状图_Python绘制堆叠柱状图的实例
时间: 2023-09-10 19:08:55 浏览: 117
好的,我来讲一下如何用 Python 绘制堆叠柱状图。
首先,我们需要导入必要的库,包括 matplotlib 和 numpy。
```python
import matplotlib.pyplot as plt
import numpy as np
```
然后,我们需要准备数据。假设我们有三个类别的数据,每个类别有四个子类别。我们可以用一个二维数组来表示这些数据。
```python
data = np.array([[15, 20, 25, 30], [10, 15, 20, 25], [5, 10, 15, 20]])
```
现在,我们可以使用 matplotlib 的 `bar` 函数来绘制堆叠柱状图。我们需要多次调用 `bar` 函数,每次传入一个参数 `bottom` 来指定当前柱状图的底部位置。每个类别的第一个子类别的底部位置都是 0,后面的子类别的底部位置则是前一个子类别的顶部位置。
```python
bottom = np.zeros(len(data[0]))
colors = ['r', 'g', 'b']
for i in range(len(data)):
plt.bar(range(len(data[i])), data[i], bottom=bottom, color=colors[i])
bottom += data[i]
```
最后,我们可以添加一些标签和标题来美化图表。
```python
plt.xticks(range(len(data[0])), ['A', 'B', 'C', 'D'])
plt.ylabel('Value')
plt.title('Stacked Bar Chart')
plt.show()
```
完整的代码如下:
```python
import matplotlib.pyplot as plt
import numpy as np
data = np.array([[15, 20, 25, 30], [10, 15, 20, 25], [5, 10, 15, 20]])
bottom = np.zeros(len(data[0]))
colors = ['r', 'g', 'b']
for i in range(len(data)):
plt.bar(range(len(data[i])), data[i], bottom=bottom, color=colors[i])
bottom += data[i]
plt.xticks(range(len(data[0])), ['A', 'B', 'C', 'D'])
plt.ylabel('Value')
plt.title('Stacked Bar Chart')
plt.show()
```
运行代码,就可以得到如下的堆叠柱状图:
![stacked_bar_chart.png](https://cdn.jsdelivr.net/gh/wizardforcel/data-science-ipython/images/matplotlib/stacked_bar_chart.png)
阅读全文