使用openpyxl的styles，實現寫入值時加背景色

所用文件、數據和上一節代碼中用的一致

本次直接貼代碼

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