对比excel，用python绘制柱状图时添加table数据表
对比excel，用python绘制柱状图时添加table数据表
我们的第88篇原创
作者：才哥
☆
大家好，我是才哥。
最近在做数据可视化的时候，希望在图上同时显示数据表。关于这个需求，用excel可以比较方便，直接快速布局中选择布局5即可。那么，如果我们想用python也来完成这项任务，可以怎么做呢？
期望效果预览：
excel作图效果
1. Excel简单绘制
就很简单了，直接选中数据插入柱状图，然后在图表工具-设计-快速布局中选择相应的布局即可。
案例数据及效果
excel图表设计->快速布局—>布局5
2. Python绘制
那这里我们用到的是matplotlib，bar和table。
将图表元素进行拆解，可以分为柱状图和数据表，刚好matplotlib提供了对应的接口。
2.1 柱状图绘制
先绘制柱状图，案例中是两组数据，所以是组合柱状图。在本次绘制中，有以下几个知识点，可以记一记：
- 设置标题时的位置（用参数
x,y指定） - 设置坐标轴标题时用参数
rotation旋转方向 - 设置坐标轴区间范围
- 显示数据标签（用
ax.bar_label方法）
import numpy as np
import matplotlib.pyplot as plt
# 中文及负数显示
plt.rcParams['font.sans-serif'] = ['Microsoft YaHei']
plt.rcParams['axes.unicode_minus'] = False
# 创建画布
fig, ax = plt.subplots(figsize=(10, 4), dpi=100)
# 案例数据
data = [[150, 200, -50, -100, -75],
[300, 125, -80, 75, -100],
]
# 列与行
columns = ('周一', '周二', '周三', '周四', '周五')
rows = ['才哥', '楠楠']
# 作图参数
index = np.arange(len(columns))-0.1
bar_width = 0.4
# 设置柱状图颜色
colors = ['turquoise', 'coral']
# 柱状图
bar1 = plt.bar(index, data[0],
bar_width, color=colors[0], edgecolor='grey')
bar2 = plt.bar(index+bar_width, data[1],
bar_width, color=colors[1], edgecolor='grey')
# 设置标题
ax.set_title('工作日收益情况', fontsize=16, y=1.1,x=0.44)
# 设置坐标轴标题
ax.set_ylabel("元", fontsize=12, color='black', alpha=0.7, rotation=360)
# 设置Y轴区间
ax.set_ylim(-150, 350)
# 显示数据标签
ax.bar_label(bar1, label_type='edge')
ax.bar_label(bar2, label_type='edge')
# x,y轴刻度不显示
ax.tick_params(axis=u'both', which=u'both', length=0)
# x轴刻度及刻度值不显示
plt.xticks([])
输出如下：
组合柱状图
2.2. 图表table绘制
先看看图表table的绘制，我们拿案例数据来单独绘制看看：
# 创建画布
fig, ax = plt.subplots(figsize=(10, 4), dpi=100)
# table图表
the_table = plt.table(cellText=data,
rowLabels=rows,rowColours=colors,
colLabels=columns, cellLoc='center', loc="bottom",
bbox=[0, -0.4, 1, 0.24]
)
输出如下：
table
关于 plt.table的参数介绍如下：
cellText：表格单元格文本，字符串中的换行符暂不支持，可能导致文本超出单元格边界cellColours：表格单元格背景色cellLoc：表格单元格文本的对齐方式，取值范围为{'left', 'center', 'right'}，默认值为'right'colWidths：表格单元格宽度rowLabels：表格行表头文本rowColours：表格行表头背景色rowLoc：表格行表头文本的对齐方式，取值范围为{'full', 'left', 'right'}，默认值为'left'colLabels：表格列表头文本colColours：表格列表头背景色colLoc：表格列表头文本的对齐方式，取值范围为{'full', 'left', 'right'}，默认值为'left'loc：单元格相对于子图的位置bbox：绘制表格的边界框
最后，我们需要做的就是将上述两个图进行组合，组合过程中可能需要做一些格式微调。
2.3. 组合
在本次案例中，对组合图需求有以下几点：
- 柱状图边框不显示
- 图表table中列名高度需要高一些，单元格的高度要低一些
- 图例位置需要和对应行一致
为了实现上诉需求，我们可以通过以下方式来处理：
# 设置单元格高度
cellDict = the_table.get_celld()
for i in range(0, len(columns)):
cellDict[(0, i)].set_height(0.6)
for j in range(1, len(rows)+1):
cellDict[(j, i)].set_height(0.4)
# 设置图表table中行名单元格的高度
cellDict[(1, -1)].set_height(0.4)
cellDict[(2, -1)].set_height(0.4)
# 设置图表table单元格文本字体
the_table.auto_set_font_size(False)
the_table.set_fontsize(10)
# 设置图表table单元格边框
for key, cell in the_table.get_celld().items():
cell.set_linewidth(0.6)
# 边框隐藏
ax.spines['top'].set_visible(False)
ax.spines['right'].set_visible(False)
ax.spines['bottom'].set_visible(False)
ax.spines['left'].set_visible(False)
name = ['', '']
# 设置图例的位置
ax.legend(name, handlelength=0.7, labelspacing=0.6,
bbox_to_anchor=(-0.1, -0.23),
loc='upper left', frameon=False)
最终，我们可以得到比较满意的效果：
最终效果
以上就是本次全部内容，大家可以修改参数多试试以熟悉掌握。