如何使用python条件为excel单元格上色，并将相同颜色的表格写回excel？

from openpyxl import Workbook
from openpyxl.styles import PatternFill

def color_cells(file_path):
    # 打开Excel文件
    wb = Workbook()
    wb = wb.load_workbook(file_path)
    sheet = wb.active

    # 定义颜色和对应的条件
    colors = {
        'FF0000FF': lambda x: x.value == 'A',  # 蓝色，如果单元格的值为'A'
        'FFFF0000': lambda x: x.value == 'B',  # 红色，如果单元格的值为'B'
        'FFFFFF00': lambda x: x.value == 'C'   # 黄色，如果单元格的值为'C'
    }

    # 遍历所有单元格，根据条件设置颜色
    for row in sheet.iter_rows():
        for cell in row:
            for color, condition in colors.items():
                if condition(cell):
                    cell.fill = PatternFill(start_color=color, end_color=color, fill_type="solid")
                    break

    # 保存修改后的Excel文件
    wb.save(file_path)

# 调用函数并传入Excel文件路径
color_cells('example.xlsx')

pip install openpyxl