使用openpyxl的styles，实现写入值时加背景色

Posted 2021-01-27 benben-wu

所用文件、数据和上一节代码中用的一致

本次直接贴代码

from openpyxl.styles import fills
from openpyxl import load_workbook
class DoExcel:
    def __init__(self,filename):
        ‘‘‘
        :param filename: excel文件名
        ‘‘‘
        self.file = filename
        self.wk = load_workbook(self.file)

    def do_excel(self,sheetname):
        ‘‘‘
        :param sheetname: 工作簿名称
        :return:
        ‘‘‘
        sheet = self.wk[sheetname]
        max_row = sheet.max_row #最大行
        max_column = sheet.max_column #最大列
        data = [] #定义一个空列表,用于存储所有数据
        for r in range(2,max_row+1):
            subdata = {} #定义一个字典,用于存储每行数据
            for c in range(1,max_column+1):
                key = sheet.cell(1,c).value  #取第一行表头数据
                subdata[key] = sheet.cell(r,c).value #字典格式，表头作为key
            data.append(subdata)
        return data
    #将返回结果回写到excel文件单元格中
    def write_back(self,sheet_name,row,col,value,color):
        ‘‘‘
        :param sheet_name: 工作簿名称
        :param row: 写入行
        :param col: 写入列
        :param value: 写入值
        :param color: 背景颜色
        :return:
        ‘‘‘
        sheet = self.wk[sheet_name]
        sheet.cell(row, col).value = value
        sheet.cell(row, col).fill = fills.GradientFill(stop=(color, color))
        self.wk.save(self.file)

if __name__ == ‘__main__‘:
    file_name = "test.xlsx"
    datas = DoExcel(file_name).do_excel(‘students‘)
    i = 2
    for data in datas:
        score = int(data.get(‘考试分数‘))
        if score >= 60:
            t_pass = ‘及格‘
            color = ‘FF0000‘
        else:
            t_pass = ‘不及格‘
            color = ‘006030‘
        DoExcel(file_name).write_back(‘students‘,i,5,t_pass,color)
        i = i+1