python xlsxwriter库生成图表的应用

 1 #!/usr/bin/env python
 2 # -*- coding:utf-8 -*-
 3 # Author:Eric.yue
 4 
 5 import xlsxwriter
 6 import random
 7 from datetime import date
 8 import collections
 9 
10 def xlwt_chart(xl_obj,table):
11 
12     #生成柱状图
13     column_chart = xl_obj.add_chart({‘type‘:‘column‘})
14     column_chart.add_series({
15         ‘name‘:‘=sheet1!$D$1‘,
16         ‘categories‘:‘=sheet1!$D$2:$D$7‘,
17         ‘values‘:‘=sheet1!$E$2:$E$7‘
18     })
19     table.insert_chart(‘G2‘,column_chart)
20 
21     #生成饼图
22     column_chart2  = xl_obj.add_chart({‘type‘:‘pie‘})
23     column_chart2.add_series({
24         ‘name‘: ‘=sheet1!$D$1‘,
25         ‘categories‘:‘=sheet1!$D$2:$D$7‘,
26         ‘values‘: ‘=sheet1!$E$2:$E$7‘
27     })
28     table.insert_chart(‘G20‘, column_chart2)
29 
30 
31 def xlwt_run():
32     data_base = [‘0-50‘,‘50-60‘,‘60-70‘,‘70-80‘,‘80-90‘,‘90-100‘]
33 
34     #生成一个有序的字典
35     chart_dict = collections.OrderedDict.fromkeys(data_base,0)
36 
37     xl_obj = xlsxwriter.Workbook(‘chart.xlsx‘)
38     table = xl_obj.add_worksheet(‘sheet1‘)
39     table.write_string(0,0,u‘姓名‘)
40     table.write_string(0,1,u‘成绩‘)
41     table.write_string(0,2,u‘日期‘)
42     table.merge_range(‘D1:E1‘, u‘成绩分布‘)
43     table.set_column(‘C:E‘,15)
44 
45     #定义格式
46     date_format = xl_obj.add_format({‘num_format‘:‘yyyy-mm-dd‘})
47     color_format = xl_obj.add_format({‘color‘:‘red‘})
48     font_format = xl_obj.add_format({‘font_color‘:‘green‘,‘bold‘:True})
49 
50     mm = 1
51     for i in xrange(1,40):
52         name = ‘name_%d‘ % i
53         score = random.randint(30,100)
54         if score <= 50:
55             chart_dict[‘0-50‘] += 1
56         elif score>50 and score<=60:
57             chart_dict[‘50-60‘] += 1
58         elif score>60 and score<=70:
59             chart_dict[‘60-70‘] += 1
60         elif score>70 and score<=80:
61             chart_dict[‘70-80‘] += 1
62         elif score>80 and score<=90:
63             chart_dict[‘80-90‘] += 1
64         else:
65             chart_dict[‘90-100‘] += 1
66 
67         if score > 60:
68             table.write_string(i, 0, name)
69             table.write_number(i, 1, score)
70         else:
71             table.write_string(i, 0, name, color_format)
72             table.write_number(i, 1, score, color_format)
73 
74         table.write_datetime(i, 2,date.today(), date_format)
75         mm = mm + 1
76 
77     #生成图表数据
78     row = 1
79     for k,v in chart_dict.items():
80         table.write_string(row, 3, k, font_format)
81         table.write_number(row, 4, v, font_format)
82         row = row+1
83 
84     xlwt_chart(xl_obj,table)
85     #使用公式
86     table.write_formula(mm,1,‘=AVERAGE(B2:B40)‘)
87     #插入带链接的图片
88     table.insert_image(‘D20‘,r‘/home/mywork/pythonchina/cto51_log/bd_logo12.png‘,{‘url‘:‘https://www.baidu.com‘})
89 
90     #关闭excel句柄
91     xl_obj.close()
92 
93 if __name__ == ‘__main__‘:
94     xlwt_run()