数据报表之Excel操作模块

Excel是当今最流行的电子表格处理软件，支持丰富的计算函数及图表，在系统运营方面广泛用于运营数据报表，比如业务质量、资源利用、安全扫描等报表，同时也是应用系统常见的文件导出格式，以便数据使用人员做进一步加工处理。本节主要讲述利用Python操作Excel的模块XlsxWriter（https://xlsxwriter.readthedocs.org），可以操作多个工作表的文字、数字、公式、图表等。XlsxWriter模块具有以下功能：

100%兼容的Excel XLSX文件，支持Excel 2003、Excel 2007等版本；

支持所有Excel单元格数据格式；

单元格合并、批注、自动筛选、丰富多格式字符串等；

支持工作表PNG、JPEG图像，自定义图表；

内存优化模式支持写入大文件。

XlsxWriter模块的安装方法如下：

# pip install XlsxWriter    #pip安装方法

# easy_install XlsxWriter    #easy_install安装方法

 

#源码安装方法

# curl -O -L http://github.com/jmcnamara/XlsxWriter/archive/master.tar.gz

# tar zxvf master.tar.gz

# cd XlsxWriter-master/

# sudo python setup.py install

下面通过一个简单的功能演示示例，实现插入文字（中英字符）、数字（求和计算）、图片、单元格格式等，代码如下：

【/home/test/XlsxWriter/simple1.py】

#coding: utf-8

import xlsxwriter

 

workbook = xlsxwriter.Workbook('demo1.xlsx')    #创建一个Excel文件

worksheet = workbook.add_worksheet()    #创建一个工作表对象

 

worksheet.set_column('A:A', 20)    #设定第一列（A）宽度为20像素

bold = workbook.add_format({'bold': True})    #定义一个加粗的格式对象

 

worksheet.write('A1', 'Hello')    #A1单元格写入'Hello'

worksheet.write('A2', 'World', bold)    #A2单元格写入'World'并引用加粗格式对象bold

worksheet.write('B2', u'中文测试', bold)    #B2单元格写入中文并引用加粗格式对象bold

 

worksheet.write(2, 0, 32)    #用行列表示法写入数字'32'与'35.5'

worksheet.write(3, 0, 35.5)    #行列表示法的单元格下标以0作为起始值，'3,0'等价于'A4'

worksheet.write(4, 0, '=SUM(A3:A4)')    #求A3:A4的和，并将结果写入'4，0'，即'A5'

 

worksheet.insert_image('B5', 'img/python-logo.png')    #在B5单元格插入图片

workbook.close()    #关闭Excel文件

程序生成的demo1.xlsx文档截图如图3-1所示。

 

更多参考 http://book.2cto.com/201411/48255.html

------------------------------------------------------------------------------------

本次实践通过定制网站5个频道的流量报表周报，通过XlsxWriter模块将流量数据写入Excel文档，同时自动计算各频道周平均流量，再生成数据图表。具体是通过workbook.add_chart({'type': 'column'})方法指定图表类型为柱形，使用write_row、write_column方法分别以行、列方式写入数据，使用add_format()方法定制表头、表体的显示风格，使用add_series()方法将数据添加到图表，同时使用chart.set_size、set_title、set_y_axis设置图表的大小及标题属性，最后通过insert_chart方法将图表插入工作表中。我们可以结合2.3节的内容来实现周报的邮件推送，本示例略去此功能。实现的代码如下：

【/home/test/XlsxWriter/simple2.py】

#coding: utf-8

import xlsxwriter

 

workbook = xlsxwriter.Workbook('chart.xlsx')    #创建一个Excel文件

worksheet = workbook.add_worksheet()    #创建一个工作表对象

chart = workbook.add_chart({'type': 'column'})    #创建一个图表对象

#定义数据表头列表

title = [u'业务名称',u'星期一',u'星期二',u'星期三',u'星期四',u'星期五',u'星期六',u'星期日',u'平均流量']

buname= [u'业务官网',u'新闻中心',u'购物频道',u'体育频道',u'亲子频道']    #定义频道名称

#定义5频道一周7天流量数据列表

data = [

    [150,152,158,149,155,145,148],

    [89,88,95,93,98,100,99],

    [201,200,198,175,170,198,195],

    [75,77,78,78,74,70,79],

    [88,85,87,90,93,88,84],

]

format=workbook.add_format()    #定义format格式对象

format.set_border(1)    #定义format对象单元格边框加粗(1像素)的格式

 

format_title=workbook.add_format()    #定义format_title格式对象

format_title.set_border(1)   #定义format_title对象单元格边框加粗(1像素)的格式

format_title.set_bg_color('#cccccc')   #定义format_title对象单元格背景颜色为

                                       #'#cccccc'的格式

format_title.set_align('center')    #定义format_title对象单元格居中对齐的格式

format_title.set_bold()    #定义format_title对象单元格内容加粗的格式

 

format_ave=workbook.add_format()    #定义format_ave格式对象

format_ave.set_border(1)    #定义format_ave对象单元格边框加粗(1像素)的格式

format_ave.set_num_format('0.00')   #定义format_ave对象单元格数字类别显示格式

 

#下面分别以行或列写入方式将标题、业务名称、流量数据写入起初单元格，同时引用不同格式对象

worksheet.write_row('A1',title,format_title)  

worksheet.write_column('A2', buname,format)

worksheet.write_row('B2', data[0],format)

worksheet.write_row('B3', data[1],format)

worksheet.write_row('B4', data[2],format)

worksheet.write_row('B5', data[3],format)

worksheet.write_row('B6', data[4],format)

 

#定义图表数据系列函数

def chart_series(cur_row):

    worksheet.write_formula('I'+cur_row, \

     '=AVERAGE(B'+cur_row+':H'+cur_row+')',format_ave)    #计算（AVERAGE函数）频

                                                          #道周平均流量

    chart.add_series({

        'categories': '=Sheet1!$B$1:$H$1',    #将“星期一至星期日”作为图表数据标签(X轴)

        'values':     '=Sheet1!$B$'+cur_row+':$H$'+cur_row,    #频道一周所有数据作

                                                               #为数据区域

        'line':       {'color': 'black'},    #线条颜色定义为black(黑色)

        'name': '=Sheet1!$A$'+cur_row,    #引用业务名称为图例项

    })

 

for row in range(2, 7):    #数据域以第2～6行进行图表数据系列函数调用

    chart_series(str(row))

 

#chart.set_table()    #设置X轴表格格式，本示例不启用

#chart.set_style(30)    #设置图表样式，本示例不启用

chart.set_size({'width': 577, 'height': 287})    #设置图表大小

chart.set_title ({'name': u'业务流量周报图表'})    #设置图表（上方）大标题

chart.set_y_axis({'name': 'Mb/s'})    #设置y轴（左侧）小标题

 

worksheet.insert_chart('A8', chart)    #在A8单元格插入图表

workbook.close()    #关闭Excel文档

上述示例将创建一个如图3-11所示的工作表。

 

 3.4.1节XlsxWrite模块的常用类与方法说明参考官网http://xlsxwriter.readthedocs.org