龙空技术网

python学习笔记 - 绘制折线图与条形图

小纯编程 171

前言:

而今兄弟们对“python读取数据画折线图”大致比较看重,同学们都想要分析一些“python读取数据画折线图”的相关内容。那么小编也在网上汇集了一些对于“python读取数据画折线图””的相关资讯,希望同学们能喜欢,小伙伴们快快来学习一下吧!

对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_workbookfrom openpyxl.chart import LineChartwb = 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_row12345678910111213141516

我们重点关注前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.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参数可以有两种方式:

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.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_axisy_axis属性。从字面信息可以看出,这是关于x轴与y轴信息的属性。通过跟踪TextAxisNumericAxis类的构造方法,可以得知它们都提供了title属性,用于调整x轴与y轴的标题,所以我们可以通过如下的方式去提调整:

# chart为折线图对象chart.x_axis.title = "季度"chart.y_axis.title = "绩效"123
1.6、设置样式并保存

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

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

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

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

2、条形图

条形图与折线图的设置方式几乎一致,区别在于生成图表对象的类不同,条形图生成图表对象的方式为:

from openpyxl.chart import BarChartchart = BarChart()12

由于绘制条形图与折线图的方式基本相同,此处不再赘述,以一个总体案例去展示条形图的生成过程。

from openpyxl import load_workbookfrom openpyxl.chart import BarChart, Referencefile_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”即可领取

标签: #python读取数据画折线图