大家好,我是才哥。
最近在做数据可视化的时候,希望在图上同时显示数据表。关于这个需求,用excel可以比较方便,直接快速布局中选择布局5
即可。那么,如果我们想用python也来完成这项任务,可以怎么做呢?
期望效果预览:
1. Excel简单绘制
就很简单了,直接选中数据插入柱状图,然后在图表工具-设计-快速布局
中选择相应的布局即可。
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]
)
输出如下:
关于 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)
最终,我们可以得到比较满意的效果:
以上就是本次全部内容,大家可以修改参数多试试以熟悉掌握。