对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

1.2、绘制数据部分

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开始。

我们希望获取以下表格中的数据:

python如何做一个折线图(python学习笔记-绘制折线图与条形图)(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.3、保存绘制好的折线图

在步骤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

1.4、设置X轴分段标签

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参数可以有两种方式:

  1. labels为Reference类型
  2. label为String类型,最终通过该String参数生成一个Reference对象。

从上述两种方式可以看出,set_categories方法中的参数就是为了生成Reference对象。而Reference对象在1.2节中也有讲过。

我们希望工作页(ws)第一行的第2、3、4、5列作为X轴的分段标签:

python如何做一个折线图(python学习笔记-绘制折线图与条形图)(2)

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.5、设置X轴与Y轴的标题

在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.6、设置样式并保存

1.1 - 1.5的步骤完成后,我们计划将得到如下完整的折线图(当然必须要保存后才能查看)。

python如何做一个折线图(python学习笔记-绘制折线图与条形图)(3)

折线图完成后,我们可以根据需求或喜好,设置它的样式。LineChart对象提供了style属性,用于设置折线图的属性。style属性的属性值取之范围为1 ~ 48,若赋值不在这个范围内则会报错。属性值设置完成后,保存工作簿。

# 设置折线图样式 chart.style = 48 # 保存工作簿 wb.save('文件路径.xlsx') 1234

最终样式获得样式如下,至于1~48值对应的样式分别代表什么样我也没试过,有兴趣的可以自行尝试。

python如何做一个折线图(python学习笔记-绘制折线图与条形图)(4)

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学习笔记-绘制折线图与条形图)(5)

小编是一名python开发工程师,这里有我自己整理了一套最新的python系统学习教程,包括从基础的python脚本到web开发、爬虫、数据分析、数据可视化、机器学习等。想要这些资料的可以关注小编,并在后台私信小编:“01”即可领取

,