Python实现在Excel文件中写入图表

学习完如何生成一个 excel 文件之后，接下来我们继续学习一下如何在 excel 文件中写入一个比较简单的图表，先来看一下所需要的几个函数（这些函数都是通过工作簿执行出来的）。

add_series 参数

$A$1:$A$4 表示从第A行的第一个值到第四个值去展示标题

图表的样式

结合 study.xlsx 文件 将学生成绩以表图形式展示

这里继续使用上一章节的 “将项目中的 study.xlsx 文件的内容写入 write.xlsx” 的案例；

代码示例如下：

# coding:utf-8

import xlsxwriter
import xlrd

# excel = xlsxwriter.Workbook('write.xlsx')       # 初始化 excel 对象
# book = excel.add_worksheet('study')             # 添加工作簿
#
# title = ['姓名', '性别', '年龄', '成绩', '等级']   # 定义 write.xlsx 的首行内容
#
# for index, data in enumerate(title):        # for循环 利用枚举函数将 title 的内容写入 "write.xlsx" 的首行
#     book.write(0, index, data)
# excel.close()

def read():             # 定义一个 read 函数读取 "study.xlsx" 文件
   result = []
   excel = xlrd.open_workbook('study.xlsx')
   book = excel.sheet_by_name('学生手册')
   for i in book.get_rows():
       content = []
       for j in i:
           content.append(j.value)
       result.append(content)
   return result

def write(content):     # 定义一个 write 函数 将读取到 "study.xlsx" 的内容写入到 "write.xlsx" 文件
   excel = xlsxwriter.Workbook('write.xlsx')
   book = excel.add_worksheet('study')

for index, data in enumerate(content):
       print(data)
       for sub_index, sub_data in enumerate(data):
           # print(sub_index, sub_data)
           book.write(index, sub_index, sub_data)

book1 = excel.add_worksheet('学生等级')     # 定义 "学生等级" 工作簿
   data = [                                   # 定义 data 列表 用以存放学生成绩所处等级与数量
       ['优秀', '良好', '中', '差'],
       [15, 25, 55, 5]
   ]

book1.write_column('A1', data[0])       # write_column() 函数，按行添加（即 A1 行添加 data 列表的 0 索引的内容）
   book1.write_column('B1', data[1])

chart = excel.add_chart({'type': 'column'})     # 定义图表样式为 柱状样式表
   chart.add_series({                              # 添加需要展示的数据
       'categories': '=学生等级!$A1:$A4',
       'values': '=学生等级!$B1:$B4',
       'name': '成绩占比'
   })
   chart.set_title({'name': '成绩占比图表'})         # 定义图表title
   book1.insert_chart('A10', chart)                # 将 chart 放在 excel 表格对应工作簿的 A10 位置

excel.close()

if __name__ == '__main__':
   result = read()
   write(result)

运行结果如下：

拓展：再试试饼状图与条形样式图

代码示例如下：

# coding:utf-8

import xlsxwriter
import xlrd

# excel = xlsxwriter.Workbook('write.xlsx')       # 初始化 excel 对象
# book = excel.add_worksheet('study')             # 添加工作簿
#
# title = ['姓名', '性别', '年龄', '成绩', '等级']   # 定义 write.xlsx 的首行内容
#
# for index, data in enumerate(title):        # for循环 利用枚举函数将 title 的内容写入 "write.xlsx" 的首行
#     book.write(0, index, data)
# excel.close()

def read():             # 定义一个 read 函数读取 "study.xlsx" 文件
   result = []
   excel = xlrd.open_workbook('study.xlsx')
   book = excel.sheet_by_name('学生手册')
   for i in book.get_rows():
       content = []
       for j in i:
           content.append(j.value)
       result.append(content)
   return result

def write(content):     # 定义一个 write 函数 将读取到 "study.xlsx" 的内容写入到 "write.xlsx" 文件
   excel = xlsxwriter.Workbook('write.xlsx')
   book = excel.add_worksheet('study')

for index, data in enumerate(content):
       print(data)
       for sub_index, sub_data in enumerate(data):
           # print(sub_index, sub_data)
           book.write(index, sub_index, sub_data)

book1 = excel.add_worksheet('学生等级')     # 定义 "学生等级" 工作簿
   data = [                                   # 定义 data 列表 用以存放学生成绩所处等级与数量
       ['优秀', '良好', '中', '差'],
       [15, 25, 55, 5]
   ]

book1.write_column('A1', data[0])       # write_column() 函数，按行添加（即 A1 行添加 data 列表的 0 索引的内容）
   book1.write_column('B1', data[1])

chart = excel.add_chart({'type': 'column'})     # 定义图表样式为 柱状样式表
   chart.add_series({                              # 添加需要展示的数据
       'categories': '=学生等级!$A1:$A4',
       'values': '=学生等级!$B1:$B4',
       'name': '成绩占比'
   })
   chart.set_title({'name': '成绩占比图表'})         # 定义图表title
   book1.insert_chart('A10', chart)                # 将 chart 放在 excel 表格对应工作簿的 A10 位置

chart = excel.add_chart({'type': 'bar'})
   chart.add_series({
       'categories': '=学生等级!$A1:$A4',
       'values': '=学生等级!$B1:$B4',
       'name': '成绩占比'
   })
   chart.set_title({'name': '成绩占比图表'})
   book1.insert_chart('A25', chart)

chart = excel.add_chart({'type': 'pie'})
   chart.add_series({
       'categories': '=学生等级!$A1:$A4',
       'values': '=学生等级!$B1:$B4',
       'name': '成绩占比'
   })
   chart.set_title({'name': '成绩占比图表'})
   book1.insert_chart('A40', chart)

excel.close()

if __name__ == '__main__':
   result = read()
   write(result)

运行结果如下：

来源：https://blog.csdn.net/weixin_42250835/article/details/124872483