Python利用pandas和matplotlib实现绘制堆叠柱状图

更新时间：2023年11月14日 08:48:37 作者：悬崖上的金鱼

在数据可视化中,堆叠柱状图是一种常用的图表类型,它能够清晰地展示多个类别的数据,本文将演示如何使用 Python 的 pandas 和 matplotlib 库绘制优化的堆叠柱状图,需要的可以参考下

在数据可视化中，堆叠柱状图是一种常用的图表类型，它能够清晰地展示多个类别的数据，并突出显示每个类别中各部分的总量和组成比例。本文将演示如何使用 Python 的 pandas 和 matplotlib 库绘制优化的堆叠柱状图，并展示了销售数量随店铺名称变化的情况。

导入必要的库

首先，我们需要导入 pandas 和 matplotlib.pyplot 库，并指定中文字体为黑体，代码如下：

import pandas as pd
import matplotlib.pyplot as plt

plt.rcParams['font.family'] = ['SimHei']  # 指定中文字体为黑体

读取数据

接下来，我们使用 pandas 的 read_excel 函数读取 Excel 文件中的数据，并指定读取的工作表名称为“Sheet3”，如下所示：

df = pd.read_excel(r'C:\Users\liuchunlin2\Desktop\新建文件夹\新建 XLSX 工作表.xlsx', sheet_name='Sheet3')

设置图形参数

在绘制堆叠柱状图之前，我们需要设置柱状图的宽度和 x 轴的位置，代码如下：

bar_width = 0.35  # 设置柱状图的宽度
x = df.index  # 设置x轴的位置

绘制堆叠柱状图

使用 matplotlib 库的 subplots 函数创建图形对象，并使用 bar 函数绘制堆叠柱状图，具体代码如下：

fig, ax = plt.subplots()
rects1 = ax.bar(x, df['销售数量'], bar_width, label='销售数量')
rects2 = ax.bar(x, df['销售数量2'], bar_width, bottom=df['销售数量'], label='销售数量2')

添加标签和标题

我们为图形添加轴标签、标题、刻度和图例，使其更具可读性，具体代码如下：

ax.set_xlabel('店铺名称')
ax.set_ylabel('销售数量')
ax.set_title('Stacked Bar Chart')
ax.set_xticks(x)
ax.set_xticklabels(df['店铺名称'])
ax.legend()

显示数据标签

最后，我们使用 annotate 函数在每个柱子上方显示数据标签，以展示具体的销售数量，具体代码如下：

for rect in rects1:
    height = rect.get_height()
    ax.annotate(f'{height}', xy=(rect.get_x() + rect.get_width() / 2, height), xytext=(0, 3),
                textcoords='offset points', ha='center', va='bottom')

for rect1, rect2 in zip(rects1, rects2):
    height1 = rect1.get_height()
    height2 = rect2.get_height()
    total_height = height1 + height2
    ax.annotate(f'{height2}', xy=(rect2.get_x() + rect2.get_width() / 2, total_height), xytext=(0, 3),
                textcoords='offset points', ha='center', va='bottom')

显示图形

最后，使用 plt.show() 函数显示绘制好的堆叠柱状图，代码如下：

plt.show()

通过以上步骤，我们成功绘制出了堆叠柱状图，展示了不同店铺的销售数量情况。

图表效果图展示

完整代码

import pandas as pd
import matplotlib.pyplot as plt

plt.rcParams['font.family'] = ['SimHei']  # 指定中文字体为黑体
# 读取Excel文件
df = pd.read_excel(r'C:\Users\liuchunlin2\Desktop\新建文件夹\新建 XLSX 工作表.xlsx', sheet_name='Sheet3')
# 设置柱状图的宽度
bar_width = 0.35
# 设置x轴的位置
x = df.index

# 绘制堆叠柱状图
fig, ax = plt.subplots()
rects1 = ax.bar(x, df['销售数量'], bar_width, label='销售数量')
rects2 = ax.bar(x, df['销售数量2'], bar_width, bottom=df['销售数量'], label='销售数量2')

# 添加标签和标题
ax.set_xlabel('店铺名称')
ax.set_ylabel('销售数量')
ax.set_title('Stacked Bar Chart')
ax.set_xticks(x)
ax.set_xticklabels(df['店铺名称'])
ax.legend()

# 在每个柱子上方显示数据标签
for rect in rects1:
    height = rect.get_height()
    ax.annotate(f'{height}', xy=(rect.get_x() + rect.get_width() / 2, height), xytext=(0, 3),
                textcoords='offset points', ha='center', va='bottom')

for rect1, rect2 in zip(rects1, rects2):
    height1 = rect1.get_height()
    height2 = rect2.get_height()
    total_height = height1 + height2
    ax.annotate(f'{height2}', xy=(rect2.get_x() + rect2.get_width() / 2, total_height), xytext=(0, 3),
                textcoords='offset points', ha='center', va='bottom')

# 显示图形
plt.show()

到此这篇关于Python利用pandas和matplotlib实现绘制堆叠柱状图的文章就介绍到这了,更多相关Python堆叠柱状图内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Django框架信号调度原理解析
这篇文章主要介绍了Django框架信号调度原理解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-09-09
wxpython实现图书管理系统
这篇文章主要为大家详细介绍了wxpython实现图书管理系统，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-03-03
Python实现读取文件夹按数字排序功能
这篇文章主要介绍了Python读取文件夹按数字排序,代码简单易懂，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2022-09-09
Python使用thread模块实现多线程的操作
线程（Threads）是操作系统提供的一种轻量级的执行单元,可以在一个进程内并发执行多个任务,每个线程都有自己的执行上下文,包括栈、寄存器和程序计数器,本文给大家介绍了Python使用thread模块实现多线程的操作,需要的朋友可以参考下
2024-10-10
Python整型运算之布尔型、标准整型、长整型操作示例
这篇文章主要介绍了Python整型运算之布尔型、标准整型、长整型操作,结合具体实例形式分析了Python中布尔型、标准整型、长整型等相关运算技巧,代码备有详尽注释,需要的朋友可以参考下
2017-07-07
python Jupyter运行时间实例过程解析
这篇文章主要介绍了python Jupyter运行时间实例过程解析,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-12-12
Django Auth应用实现用户身份认证
Django Auth 应用一般用在用户的登录注册上，用于判断当前的用户是否合法。本文将介绍Auth的另一个功能，即认证用户身份，感兴趣的同学可以关注一下
2021-12-12
python tarfile压缩包操作保姆级教程
TarFile类就是tar压缩包，其由member块组成, member块则包括header块和data块. 每个member以TarInfo对象形式描述. 所以TarFile就是TarInfo的序列
2023-02-02
Python制作基础学生信息管理系统
本文详细讲解了Python制作基础学生信息管理系统的实现，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-12-12
Python Pytorch深度学习之数据加载和处理
今天小编就为大家分享一篇Pytorch 数据加载与数据预处理方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2021-10-10