1 import xlrd, xlwt 2 3 #读取excel文件 4 def read_excel(url):#传入源文件读取路径 5 # 获取数据 6 data = xlrd.open_workbook(url) 7 # 获取sheet 8 # table = data.sheet_by_name(sheet_name) #通过sheet名称获取sheet数据 9 table = data.sheet_by_index(0) #通过sheet索引获取sheet数据 10 # 获取总行数 11 nrows = table.nrows 12 # 获取总列数 13 ncols = table.ncols 14 # 获取一个单元格的数值 15 cell_value = table.cell(0, 1).value #获取第0行第1列的值，即 B1 的值 16 print(cell_value) 17 18 #写入excel文件 19 def write_excel(url, sheet_name, data): #传入文件存储路径、excel的sheet名称、以及要插入的数据 20 #data = ((‘biqi‘, 963, 177), (‘editor_Intern1‘, 912, 154), (‘editor_Intern10‘, 840, 163), (‘editor_Intern11‘, 644, 173)) #模板数据 21 myWorkbook = xlwt.Workbook() #创建excel 22 mySheet = myWorkbook.add_sheet(sheet_name) #创建excel中的sheet 23 for i, val in enumerate(data): 24 for j, value in enumerate(val): 25 mySheet.write(i, j, value) #遍历数据插入sheet中 26 myWorkbook.save(url) #将创建的excel保存在该路径下
1 import xlsxwriter
2
3 #写入excel文件
4 def write_excel(url): #传入excel存储路径
5 #将要插入excel并用作绘图的数据
6 data_all = {‘错误汇总‘: {
7 ‘error_summary_header‘: [‘账号‘, ‘姓名‘, ‘投资评级错误‘, ‘评级变化错误‘, ‘股票代码错误‘, ‘股票名称错误‘, ‘目标价错误‘, ‘目标价高错误‘, ‘净利润错误‘,
8 ‘归母净利润错误‘, ‘年份错误‘, ‘分析师名字错误‘, ‘邮箱错误‘, ‘证券职业编码错误‘, ‘电话错误‘, ‘记录错误数‘, ‘错误数（人工）‘,
9 ‘错误数（研报本身）‘, ‘错误数（抽查）‘, ‘总记录数‘, ‘错误率‘, ‘研报总数‘],
10 ‘editor_Intern1‘: [‘editor_Intern1‘, ‘editor_Intern1‘, 0, 0, 0, 0, 0, 0, 0, 2, 0, 0, 0, 0, 0, 2, 0, 0, 0, 912, ‘0.22%‘,
11 223],
12 ‘editor_Intern10‘: [‘editor_Intern10‘, ‘editor_Intern10‘, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 837, ‘0.12%‘,
13 232],
14 ‘editor_Intern11‘: [‘editor_Intern11‘, ‘editor_Intern11‘, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 644, ‘0.16%‘,
15 173],
16 ‘editor_Intern12‘: [‘editor_Intern12‘, ‘editor_Intern12‘, 0, 0, 0, 0, 0, 0, 0, 1, 0, 0, 0, 0, 0, 1, 0, 0, 0, 439, ‘0.23%‘,
17 99],
18 ‘editor_Intern2‘: [‘editor_Intern2‘, ‘editor_Intern2‘, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1132, ‘0.0%‘,
19 265],
20 ‘editor_Intern3‘: [‘editor_Intern3‘, ‘editor_Intern3‘, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 852, ‘0.0%‘, 280],
21 ‘editor_Intern4‘: [‘editor_Intern4‘, ‘editor_Intern4‘, 0, 0, 0, 0, 0, 0, 7, 0, 0, 0, 0, 0, 0, 7, 0, 0, 0, 1002, ‘0.7%‘,
22 297],
23 ‘editor_Intern5‘: [‘editor_Intern5‘, ‘editor_Intern5‘, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 571, ‘0.0%‘,
24 213],
25 ‘editor_Intern6‘: [‘editor_Intern6‘, ‘editor_Intern6‘, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 778, ‘0.0%‘,
26 200],
27 ‘editor_Intern8‘: [‘editor_Intern8‘, ‘editor_Intern8‘, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
28 ‘editor_Intern9‘: [‘editor_Intern9‘, ‘editor_Intern9‘, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 0, 0, 0, 5, 0, 0, 0, 762, ‘0.66%‘, -3],
29 ‘editor_Intern13‘: [‘editor_Intern13‘, ‘editor_Intern13‘, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 729, ‘0.0%‘,
30 180],
31 ‘editor_Intern14‘: [‘editor_Intern14‘, ‘editor_Intern14‘, 0, 0, 0, 0, 0, 0, 0, 0, 8, 0, 2, 0, 0, 10, 0, 0, 0, 467, ‘2.14%‘,
32 102],
33 ‘Total‘: [‘Total‘, ‘合计‘, 0, 0, 0, 0, 0, 0, 12, 3, 9, 0, 4, 0, 0, 28, 0, 0, 0, 10102, ‘0.28%‘, 2595]}}
34 # 数据表头
35 error_detail_header = [‘研报_id‘, ‘账号‘, ‘投资评级错误‘, ‘评级变化错误‘, ‘股票代码错误‘, ‘股票名称错误‘, ‘目标价错误‘, ‘目标价高错误‘, ‘净利润错误‘, ‘归母净利润错误‘,
36 ‘年份错误‘, ‘分析师名字错误‘, ‘邮箱错误‘, ‘证券职业编码错误‘, ‘电话错误‘]
37 #创建excel
38 myWorkbook = xlsxwriter.Workbook(url)
39 # 自定义样式
40 bold = myWorkbook.add_format({
41 ‘font_size‘: 10, # 字体大小
42 ‘bold‘: True, # 是否粗体
43 ‘bg_color‘: ‘#101010‘, # 表格背景颜色
44 ‘font_color‘: ‘#FEFEFE‘, # 字体颜色
45 ‘align‘: ‘center‘, # 居中对齐
46 ‘top‘: 2, # 上边框
47 ‘left‘: 2, # 左边框
48 ‘right‘: 2, # 右边框
49 ‘bottom‘: 2 # 底边框
50 })
51 for k, v in data_all.items():
52 if k == ‘错误明细‘:
53 mySheet1 = myWorkbook.add_worksheet(k) #创建“错误明细”sheet
54 for index, header in enumerate(error_detail_header):
55 mySheet1.write(0, index, header, bold)
56 for i, val in enumerate(v):
57 i += 1
58 for j, value in enumerate(val):
59 mySheet1.write(i, j, value, bold) #向第i行第j列插入数据，并使用bold定义的样式
60 if k == ‘错误汇总‘:
61 mySheet2 = myWorkbook.add_worksheet(k)
62 i = 0
63 for summary_value in v.values():
64 for sum_index, sum_value in enumerate(summary_value):
65 mySheet2.write(i, sum_index, sum_value, bold)
66 i += 1
67
68 ‘‘‘绘制错误数柱状图‘‘‘
69 # 创建一个柱状图(column chart)
70 chart_col = myWorkbook.add_chart({‘type‘: ‘column‘})
71
72 # 图表下方显示数据表格
73 chart_col.set_table({
74 ‘show_keys‘: True
75 })
76
77 # 配置数据(用了另一种语法)
78 chart_col.add_series({
79 ‘name‘: ‘=错误汇总!$P$1‘,
80 ‘categories‘: ‘=错误汇总!$B$2:$B$14‘,
81 ‘values‘: ‘=错误汇总!$P$2:$P$14‘,
82 ‘line‘: {‘color‘: ‘#C0504D‘},
83 ‘fill‘: {‘color‘: ‘#C0504D‘},
84 ‘data_labels‘: {‘value‘: True}, #在图表上显示对应的数据
85 })
86
87 # # 配置数据
88 # chart_col.add_series({
89 # ‘name‘: [‘错误汇总‘, 0, 2],
90 # ‘categories‘: [‘错误汇总‘, 1, 0, 6, 0],
91 # ‘values‘: [‘错误汇总‘, 1, 2, 6, 2],
92 # ‘line‘: {‘color‘: ‘red‘},
93 # })
94
95 # 设置图表的title 和 x，y轴信息
96 chart_col.set_title({‘name‘: ‘记录错误数‘})
97 chart_col.set_x_axis({‘name‘: ‘员工‘})
98 chart_col.set_y_axis({‘name‘: ‘错误数‘})
99
100 # 设置图表的风格
101 chart_col.set_style(10)
102
103 # 把图表插入到worksheet以及偏移
104 mySheet2.insert_chart(‘A21‘, chart_col, {
105 ‘x_offset‘: 0,
106 ‘y_offset‘: 0,
107 ‘x_scale‘: 1.5,
108 ‘y_scale‘: 1.5,
109 }) #第一个参数为图表插入的起始位置， x_offset、y_offset为偏移量， x_scale、y_scale为缩放比率
110
111 ‘‘‘绘制错误率折线图‘‘‘
112 chart_col1 = myWorkbook.add_chart({‘type‘: ‘line‘})
113 # 配置数据(用了另一种语法)
114 chart_col1.add_series({
115 ‘name‘: ‘=错误汇总!$U$1‘,
116 ‘categories‘: ‘=错误汇总!$B$2:$B$14‘,
117 ‘values‘: ‘=错误汇总!$U$2:$U$14‘,
118 ‘line‘: {‘color‘: ‘#C0504D‘},
119 ‘data_labels‘: {‘value‘: True},
120 })
121
122 # # 配置数据
123 # chart_col1.add_series({
124 # ‘name‘: [‘错误汇总‘, 0, 2],
125 # ‘categories‘: [‘错误汇总‘, 1, 0, 6, 0],
126 # ‘values‘: [‘错误汇总‘, 1, 2, 6, 2],
127 # ‘line‘: {‘color‘: ‘red‘},
128 # })
129
130 # 设置图表的title 和 x，y轴信息
131 chart_col1.set_title({‘name‘: ‘错误率‘})
132 chart_col1.set_x_axis({‘name‘: ‘员工‘})
133 chart_col1.set_y_axis({‘name‘: ‘错误数‘})
134
135 # 设置图表的风格
136 chart_col1.set_style(1)
137
138 # 将柱状图合并入折线图中
139 # chart_col1.combine(chart_col)
140
141 mySheet2.insert_chart(‘N21‘, chart_col1, {
142 ‘x_offset‘: 0,
143 ‘y_offset‘: 0,
144 ‘x_scale‘: 1.5,
145 ‘y_scale‘: 1.5,
146 }) #第一个参数为图表插入的起始位置， x_offset、y_offset为偏移量， x_scale、y_scale为缩放比率
147
148 ‘‘‘绘制错误种类饼图‘‘‘
149 # 创建一个柱状图(column chart)
150 chart_col2 = myWorkbook.add_chart({‘type‘: ‘pie‘})
151 # 配置数据(用了另一种语法)
152 chart_col2.add_series({
153 ‘categories‘: ‘=错误汇总!$C$1:$O$1‘,
154 ‘values‘: ‘=错误汇总!$C$15:$O$15‘,
155 ‘data_labels‘: {‘value‘: True},
156 ‘points‘: [
157 {‘fill‘: {‘color‘: ‘#4590A7‘}},
158 {‘fill‘: {‘color‘: ‘#AA4643‘}},
159 {‘fill‘: {‘color‘: ‘#89A54E‘}},
160 {‘fill‘: {‘color‘: ‘#71588F‘}},
161 {‘fill‘: {‘color‘: ‘#4198AF‘}},
162 {‘fill‘: {‘color‘: ‘#DB843D‘}},
163 {‘fill‘: {‘color‘: ‘#93A9CF‘}},
164 {‘fill‘: {‘color‘: ‘#D19392‘}},
165 {‘fill‘: {‘color‘: ‘#B9CD96‘}},
166 {‘fill‘: {‘color‘: ‘#4590A7‘}},
167 {‘fill‘: {‘color‘: ‘#AA4643‘}},
168 {‘fill‘: {‘color‘: ‘#89A54E‘}},
169 {‘fill‘: {‘color‘: ‘#71588F‘}},
170 {‘fill‘: {‘color‘: ‘#4198AF‘}},
171 {‘fill‘: {‘color‘: ‘#DB843D‘}},
172 {‘fill‘: {‘color‘: ‘#93A9CF‘}},
173 {‘fill‘: {‘color‘: ‘#D19392‘}},
174 {‘fill‘: {‘color‘: ‘#B9CD96‘}},
175 ] #饼状图会使用到的色号
176 })
177
178 # # 配置数据
179 # chart_col.add_series({
180 # ‘name‘: [‘错误汇总‘, 0, 2],
181 # ‘categories‘: [‘错误汇总‘, 1, 0, 6, 0],
182 # ‘values‘: [‘错误汇总‘, 1, 2, 6, 2],
183 # ‘line‘: {‘color‘: ‘red‘},
184 # })
185
186 # 设置图表的title
187 chart_col2.set_title({‘name‘: ‘错误种类占比‘})
188
189 # 设置图表的风格
190 chart_col2.set_style(1)
191
192 # 把图表插入到worksheet以及偏移
193 mySheet2.insert_chart(‘H45‘, chart_col2, {
194 ‘x_offset‘: 0,
195 ‘y_offset‘: 0,
196 ‘x_scale‘: 1.5,
197 ‘y_scale‘: 1.5,
198 }) #第一个参数为图表插入的起始位置， x_offset、y_offset为偏移量， x_scale、y_scale为缩放比率
199
200 myWorkbook.close()
201
202 if __name__ == ‘__main__‘:
203 write_excel(‘./datasource/test_query_bak.xls‘)
总体来说，如果只需要从excel读取数据及向excel中插入数据，用xlrd包及xlwt包就可以很好的解决了；但如果想要在excel中绘制各种报表图片就需要借助xlsxwriter包来进行了（xlsxwriter官网：http://xlsxwriter.readthedocs.io/chart.html）
附上xlsxwriter绘制的图表的截图
原文地址：https://www.cnblogs.com/Goongo/p/9383138.html
时间： 2024-10-13 13:40:18