Python Pandas：样式列标题

Question

我正在使用Pandas 样式器根据列标题的名称为某些列提供背景颜色。虽然这按预期工作，但列标题的背景颜色不会改变。这是我的脚本中应用了你的风格的部分：def highlight_col(x):    if x.name in added_columns:        return ['background-color: #67c5a4']*x.shape[0]    elif x.name in dropped_columns:        return ['background-color: #ff9090']*x.shape[0]    else:        return ['background-color: None']*x.shape[0]old = old.style.apply(highlight_col, axis=0)有没有办法将style.apply() 函数不仅应用于列标题下方的单元格，还应用于包括列标题在内的整个列？编辑：为了澄清这里是 excel 输出的 屏幕截图： excel 输出的屏幕截图“标题 2”应该与它下面的单元格具有相同的背景颜色。

Answer 1

好的，我想我想出了一种使用 html 'selectors' 处理列标题格式的方法：

使用您的大部分代码作为设置：

df = pd.DataFrame('some value', columns=['Header1','Header2','Header3'], index=np.arange(12))

added_columns = 'Header2'

dropped_columns = 'Header1'

def highlight_col(x):

    if x.name in added_columns:

        return ['background-color: #67c5a4']*x.shape[0]

    elif x.name in dropped_columns:

        return ['background-color: #ff9090']*x.shape[0]

    else:

        return ['background-color: None']*x.shape[0]

col_loc_add = df.columns.get_loc(added_columns) + 2

col_loc_drop = df.columns.get_loc(dropped_columns) + 2

df.style.apply(highlight_col, axis=0)\

  .set_table_styles(

     [{'selector': f'th:nth-child({col_loc_add})',

       'props': [('background-color', '#67c5a4')]},

     {'selector': f'th:nth-child({col_loc_drop})',

       'props': [('background-color', '#ff9090')]}])

输出：

注意：我使用的是 Python 3.6+ 特性的 f-string。

Answer 2

您可以使用np.vstack()如下堆叠列名并创建一个新的数据框来应用该函数，然后使用以下命令导出到 excel header=False：

使用@Scott 的数据和 piR 的函数，设置：

df = pd.DataFrame('some value', columns=['Header1','Header2','Header3'], index=np.arange(12))

def f(dat, c='red'):

    return [f'background-color: {c}' for i in dat]

你可以做：

pd.DataFrame(np.vstack((df.columns,df.to_numpy())),columns=df.columns).style.apply(

                  f,subset=['Header2']).to_excel('file.xlsx',header=False,index=False)

excel文件输出：