代码示例:
# !usr/bin/env python
# -*- coding:utf-8 -*-
'''
function:设计excel表格
author:AI算法联盟
date:20220401
'''
''' 若未安装openpyxl模块,先执行pip install openpyxl进行安装'''
import pandas as pd
from openpyxl import Workbook
from openpyxl.styles import Border, Side, Font, Alignment
from openpyxl.styles import PatternFill, Color
def get_data():
data = [["高一", 529, 797, 1326],
["高二", 502, 815, 1317],
["高三", 671, 749, 1420],
["总计", 1702, 2361, 4063]]
data_df = pd.DataFrame(data, columns=["年级", "男生", "女生", "总人数"])
return data_df
def excel_design(data_df):
columns_num = len(data_df.columns)
wb = Workbook()
ws = wb[wb.sheetnames[0]] # 选择第一张表
ws.title = "demo" # 工作表重命名
# 单元格背景颜色设计
color_head = PatternFill("solid",