对openpyxl中【工作簿】、【工作页】、【行 / 列】与【单元格】的概念存在疑惑的可以参考《Excel工作簿、工作页、行列、单元格介绍》。
1、折线图折线图有着可以显示随时间(根据常用比例设置)而变化的连续数据等优点,因此非常适用于显示在相等时间间隔下数据的趋势。我们可以将排列在工作表的列或行中的数据绘制到折线图中。在折线图中,类别数据沿水平轴均匀分布,所有值数据沿垂直轴均匀分布。
1.1、创建折线图对象对于面向对象的角度而言,我们需要绘制一个折线图,所以需要创建一个可以绘制折线图的对象,去辅助我们绘制折线图。openpyxl库下的chart模块为我们提供了LineChart类,去构造我们需要的折线图对象。我们首先观察一下LineChart类的构造方法:
def __init__(self, hiLowLines=None, upDownBars=None,
marker=None, smooth=None, extLst=None, **kw):
self.hiLowLines = hiLowLines
self.upDownBars = upDownBars
self.marker = marker
self.smooth = smooth
self.x_axis = TextAxis()
self.y_axis = NumericAxis()
super(LineChart, self).__init__(**kw)
123456789
其中没有我们必要初始化的参数,所以我们可以直接空参构造一个折线图对象。
from openpyxl import load_workbook
from openpyxl.chart import LineChart
wb = load_workbook('文件路径.xlsx')
ws = wb.active
# 构建LineChart对象
chart = LineChart()
1234567
openpyxl库下的chart包提供了add_data方法,可帮助用户自动绘制折线图,那么如何使用该方法呢,首先我们先看一下该方法的源码:
def add_data(self, data, from_rows=False, titles_from_data=False):
if not isinstance(data, Reference):
data = Reference(range_string=data)
# 后续代码略...
1234
我们来看下它对应的参数:
参数说明介绍data数据部分对象可以通过后续的方法得知,该参数要么是一个Reference对象,要么用来生成Reference对象。from_rows是否以行读取数据布尔类型,默认为false,即以列读取数据。titles_from_data第一列是否标题列布尔类型,默认为false,即第一列直接开始作为数据列,而非标题列。
以上参数中,最重要的参数data为Reference对象,所以我们需要实例化一个Reference对象,为data参数赋值,我们首先来看一下Reference类的构造方法:
def __init__(self, worksheet=None, min_col=None, min_row=None,
max_col=None, max_row=None, range_string=None):
if range_string is not None:
sheetname, boundaries = range_to_tuple(range_string)
min_col, min_row, max_col, max_row = boundaries
worksheet = DummyWorksheet(sheetname)
self.worksheet = worksheet
self.min_col = min_col
self.min_row = min_row
if max_col is None:
max_col = min_col
self.max_col = max_col
if max_row is None:
max_row = min_row
self.max_row = max_row
12345678910111213141516
我们重点关注前5个参数:
参数说明介绍worksheet当前工作页操作的当前工作页对象。min_col起始列表示从第几列开始获取,下标以1开始。min_row起始行表示从第几行开始获取,下标以1开始。max_col结束列表示在第几列结束获取,下标以1开始。max_row结束行表示在第几行结束获取,下标以1开始。
我们希望获取以下表格中的数据:
chart = LineChart()
data = Reference(worksheet=ws, min_row=2, max_row=6, min_col=1, max_col=5)
chart.add_data(data, from_rows=True, titles_from_data=True)
123
在步骤1.2中,折线图的数据部分已经绘制完成了,我们需要将这个折线图保存到Sheet工作页中。openpyxl中工作页类(<class 'openpyxl.worksheet.worksheet.Worksheet'>)提供了add_chart方法,用于将生成的图表放到指定的单元格内,我们看一下这个方法:
def add_chart(self, chart, anchor=None):
if anchor is not None:
chart.anchor = anchor
self._charts.append(chart)
1234
参数说明介绍chart图表对象已经绘制好的图表对象,如折线图:chart = LineChart()anchor坐标图表存放的具体单元格坐标。
将生成完的折线图保存到Sheet工作页中的【F1】单元格中
ws.add_chart(chart, 'F1')
1
X轴分段常用来区分各个区段的数据,常见如时间、地域等。LineChart类中提供了set_categories方法,用于设定X轴各个分段的数据,我们首先看一下set_categories方法:
def set_categories(self, labels):
if not isinstance(labels, Reference):
labels = Reference(range_string=labels)
for s in self.ser:
s.cat = AxDataSource(numRef=NumRef(f=labels))
12345
在该方法中可以看出,labels参数可以有两种方式:
- labels为Reference类型
- label为String类型,最终通过该String参数生成一个Reference对象。
从上述两种方式可以看出,set_categories方法中的参数就是为了生成Reference对象。而Reference对象在1.2节中也有讲过。
我们希望工作页(ws)第一行的第2、3、4、5列作为X轴的分段标签:
cats = Reference(worksheet=ws, min_col=2, max_col=5, min_row=1, max_row=1)
# chart为 chart = LineChart() 产生的对象
chart.set_categories(cats)
123
在1.1中的构造方法中我们可以看到这两条代码:
def __init__(self, hiLowLines=None, upDownBars=None,
marker=None, smooth=None, extLst=None, **kw):
# ...
self.x_axis = TextAxis()
self.y_axis = NumericAxis()
# ...
123456
从该构造方法中的赋值可见,LineChart对象包含了x_axis与y_axis属性。从字面信息可以看出,这是关于x轴与y轴信息的属性。通过跟踪TextAxis与NumericAxis类的构造方法,可以得知它们都提供了title属性,用于调整x轴与y轴的标题,所以我们可以通过如下的方式去提调整:
# chart为折线图对象
chart.x_axis.title = "季度"
chart.y_axis.title = "绩效"
123
1.1 - 1.5的步骤完成后,我们计划将得到如下完整的折线图(当然必须要保存后才能查看)。
折线图完成后,我们可以根据需求或喜好,设置它的样式。LineChart对象提供了style属性,用于设置折线图的属性。style属性的属性值取之范围为1 ~ 48,若赋值不在这个范围内则会报错。属性值设置完成后,保存工作簿。
# 设置折线图样式
chart.style = 48
# 保存工作簿
wb.save('文件路径.xlsx')
1234
最终样式获得样式如下,至于1~48值对应的样式分别代表什么样我也没试过,有兴趣的可以自行尝试。
2、条形图
条形图与折线图的设置方式几乎一致,区别在于生成图表对象的类不同,条形图生成图表对象的方式为:
from openpyxl.chart import BarChart
chart = BarChart()
12
由于绘制条形图与折线图的方式基本相同,此处不再赘述,以一个总体案例去展示条形图的生成过程。
from openpyxl import load_workbook
from openpyxl.chart import BarChart, Reference
file_path = './测试报表.xlsx'
wb = load_workbook(file_path)
ws = wb.active
# 创建条形图对象
chart = BarChart()
# 绘制数据部分
data = Reference(worksheet=ws, min_row=2, max_row=6, min_col=1, max_col=5)
chart.add_data(data, from_rows=True, titles_from_data=True)
# 放置绘制好的柱状图
ws.add_chart(chart, 'F1')
# 设置X轴分段标签
cats = Reference(worksheet=ws, min_row=1, max_row=1, min_col=2, max_col=5)
chart.set_categories(cats)
# 设置X轴与Y轴的标题
chart.x_axis.title = '季度'
chart.y_axis.title = '绩效'
wb.save(file_path)
123456789101112131415161718192021
小编是一名python开发工程师,这里有我自己整理了一套最新的python系统学习教程,包括从基础的python脚本到web开发、爬虫、数据分析、数据可视化、机器学习等。想要这些资料的可以关注小编,并在后台私信小编:“01”即可领取
,