python實現對excel中需要的數據的單元格填充顏色

import openpyxl
from openpyxl import load_workbook

# 比如打開test.xlsx
wb = load_work(filename='test.xlsx')
# 使用第一個sheet作為工作簿
work = wb[wb.sheetnames[0]]

import openpyxl
from openpyxl.styles import PatternFill

#根據上面的work進行單元格選擇
# 設置樣式(填充色)
# 顏色必須使用hex 十六進制並且沒有'#'符號 列舉為黃色
fill = PatternFill('solid',fgColor='FFFF00')
# 填充D4為為黃色
d4 = work['D4']
d4.fill = fill

import math
import string

def cycle_letter(arr,level):
    import string
    list1 = string.ascii_uppercase
    tempArr = []
    letterArr = [i for i in list1]
    arrNum = len(arr)
    if(level==0 or arrNum==0):
        return letterArr
    for index in range(arrNum):
        for letter in letterArr:
            tempArr.append(arr[index]+letter)
    return tempArr
def reduce_excel_col_name(num):
    tempVal = 1
    level = 1
    while(tempVal):
        tempVal = num/(math.pow(26, level))
        if(tempVal>1):
            level += 1
        else:
            break
    excelArr = []
    tempArr = []
    for index in range(level):
        tempArr = cycle_letter(tempArr,index)
        for numIndex in range(len(tempArr)): 
            if(len(excelArr)<num):
                excelArr.append(tempArr[numIndex])
            else:
                return excelArr
    return excelArr

# 導入所需的庫
import math
import string
import openpyxl
import pandas as pd
from openpyxl import load_workbook

# 如上圖,一共10列,從0到9
# 產生為10的excel對應的列
def cycle_letter(arr,level):
    list1 = string.ascii_uppercase
    tempArr = []
    letterArr = [i for i in list1]
    arrNum = len(arr)
    if(level==0 or arrNum==0):
        return letterArr
    for index in range(arrNum):
        for letter in letterArr:
            tempArr.append(arr[index]+letter)
    return tempArr
 
def reduce_excel_col_name(num):
    tempVal = 1
    level = 1
    while(tempVal):
        tempVal = num/(math.pow(26, level))
        if(tempVal>1):
            level += 1
        else:
            break
    excelArr = []
    tempArr = []
    for index in range(level):
        tempArr = cycle_letter(tempArr,index)
        for numIndex in range(len(tempArr)): 
            if(len(excelArr)<num):
                excelArr.append(tempArr[numIndex])
            else:
                return excelArr
    return excelArr
# 保存到charter_list
charter_list = reduce_excel_col_name(10)

# 循環data選出所有>50的數字
# 並取出index和col_index並保存到excel中
a = []
b = []
for i in range(len(data.values)):
    for y in data.values[i]:
        if y > 50:
            # 獲取行數
            a.append(i)
            # 獲取列數
            a.append(data.values[i].tolist().index(y))
            b.append(a)
            a = []
# 列表中第一個代表的是行,第二個數字代表的是列
# 將第二個數使用charter_list產生的代替
# 因為列表的第二個數對應的也是charter_list的index
for i in b:
    i[1] = charter_list[i[1]]
    
# 這裡需要註意的是excel都是從1開始,所以我們的第一個數
# 行數需要+1,但是因為有列名0,1,2,3到9的存在,所以需要添加一個2
for i in b:
    i[0] = i[0]+2

# 將列表變為'A2','C2',符合excle的習慣
list_color_all = [x[1]+str(x[0]) for x in b]

# 使用load_workbook加載excel數據,進行顏色填充
wb = load_workbook(filename='excel_col.xlsx')
work = wb[wb.sheetnames[0]]
fill = PatternFill("solid", fgColor='FFFF00')

for i in list_color_all:
    work[i].fill = fill
wb.close()
wb.save('excel_col.xlsx')