import pandas as pd
import numpy as np
 
columns = [['A', 'A', 'B', 'B', 'C'], ['a', 'b', 'c', 'd', 'e']]
# 創(chuàng)建形狀為（10，5） 的DataFrame 并設(shè)置二級標(biāo)題
demo_df = pd.DataFrame(np.arange(50).reshape(10, 5), columns=columns)
print(demo_df)
 
def style_color(df, colors):
  """
  
  :param df: pd.DataFrame
  :param colors: 字典 內(nèi)容是 {標(biāo)題:顏色}
  :return: 
  """
  return df.style.apply(style_apply, colors=colors)
 
def style_apply(series, colors, back_ground=''):
  """
  :param series: 傳過來的數(shù)據(jù)是DataFramt中的一列  類型為pd.Series
  :param colors: 內(nèi)容是字典 其中key 為標(biāo)題名  value 為顏色
  :param back_ground: 北京顏色
  :return:
  """
  series_name = series.name[0]
  a = list()
  # 為了給每一個單元格上色
  for col in series:
    # 其中 col 為pd.DataFrame 中的 一個小單元格  大家可以根據(jù)不同需求為單元格設(shè)置不同的顏色
    # 獲取什么一級標(biāo)題獲取什么顏色
    if series_name in colors:
      for title_name in colors:
        if title_name == series_name:
          back_ground = 'background-color: ' + colors[title_name]
          # '; border-left-color: #080808'
    a.append(back_ground)
  return a
 
style_df = style_color(demo_df, {"A": '#1C1C1C', "B": '#00EEEE', "C": '#1A1A1A'})
 
with pd.ExcelWriter('df_style.xlsx', engine='openpyxl') as writer:
  #注意： 二級標(biāo)題的to_excel index 不能為False
  style_df.to_excel(writer, sheet_name='sheet_name')

quantity_year.style.applymap(lambda v
               : 'background-color: %s' %'#FFCCFF' if v>quantity_year.mean().mean() 
               else'background-color: %s'% '')