python xlsxwriter庫生成圖表的應用示例

python xlsxwriter庫生成圖表的應用示例

xlsxwriter可能用過的人並不是很多,不過使用後就會感覺,他的功能讓你歎服,除了可以按要求生成你所需要的excel外

還可以加上很形象的各種圖,比如柱狀圖、餅圖、折線圖等。

xlsxwriter 基本用法,建立 xlsx 檔案並新增資料

官方文件:http://xlsxwriter.readthedocs.org/  

xlsxwriter 可以操作 xls 格式檔案

注意:xlsxwriter 只能建立新檔案,不可以修改原有檔案。如果建立新檔案時與原有檔案同名,則會覆蓋原有檔案

Linux 下安裝: sudo pip install XlsxWriter

Windows 下安裝: pip install XlsxWriter

請看本人生成的:

 

這裡包含了資料公式的計算,插入圖片的連線,生成的圖表,當然如果你還需要其他的功能,可以繼續參考庫的文件

下面把原始碼貼出來,希望對大家在工作遇到類似的情況,可以直接拿去用。


#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author:Eric.yue
import xlsxwriter
import random
from datetime import date
import collections
def xlwt_chart(xl_obj,table):
#生成柱狀圖
column_chart = xl_obj.add_chart({'type':'column'})
column_chart.add_series({
'name':'=sheet1!$D$1',
'categories':'=sheet1!$D$2:$D$7',
'values':'=sheet1!$E$2:$E$7'
})
table.insert_chart('G2',column_chart)
#生成餅圖
column_chart2 = xl_obj.add_chart({'type':'pie'})
column_chart2.add_series({
'name': '=sheet1!$D$1',
'categories':'=sheet1!$D$2:$D$7',
'values': '=sheet1!$E$2:$E$7'
})
table.insert_chart('G20', column_chart2)
def xlwt_run():
data_base = ['0-50','50-60','60-70','70-80','80-90','90-100']
#生成一個有序的字典
chart_dict = collections.OrderedDict.fromkeys(data_base,0)
xl_obj = xlsxwriter.Workbook('chart.xlsx')
table = xl_obj.add_worksheet('sheet1')
table.write_string(0,0,u'姓名')
table.write_string(0,1,u'成績')
table.write_string(0,2,u'日期')
table.merge_range('D1:E1', u'成績分佈')
table.set_column('C:E',15)
#定義格式
date_format = xl_obj.add_format({'num_format':'yyyy-mm-dd'})
color_format = xl_obj.add_format({'color':'red'})
font_format = xl_obj.add_format({'font_color':'green','bold':True})
mm = 1
for i in xrange(1,40):
name = 'name_%d' % i
score = random.randint(30,100)
if score <= 50:
chart_dict['0-50']  = 1
elif score>50 and score<=60:
chart_dict['50-60']  = 1
elif score>60 and score<=70:
chart_dict['60-70']  = 1
elif score>70 and score<=80:
chart_dict['70-80']  = 1
elif score>80 and score<=90:
chart_dict['80-90']  = 1
else:
chart_dict['90-100']  = 1
if score > 60:
table.write_string(i, 0, name)
table.write_number(i, 1, score)
else:
table.write_string(i, 0, name, color_format)
table.write_number(i, 1, score, color_format)
table.write_datetime(i, 2,date.today(), date_format)
mm = mm   1
#生成圖表資料
row = 1
for k,v in chart_dict.items():
table.write_string(row, 3, k, font_format)
table.write_number(row, 4, v, font_format)
row = row 1
xlwt_chart(xl_obj,table)
#使用公式
table.write_formula(mm,1,'=AVERAGE(B2:B40)')
#插入帶連結的圖片
table.insert_image('D20',r'/home/mywork/pythonchina/cto51_log/bd_logo12.png',{'url':'https://www.baidu.com'})
#關閉excel控制代碼
xl_obj.close()
if __name__ == '__main__':
xlwt_run()

沒有使用類寫,只是即興而作。