Python Pandas：样式列标题

Question

我正在使用pandas styler 根据列标题的名称为某些列提供背景颜色。 虽然这按预期工作，但列标题的背景颜色不会改变。

这是我的脚本中应用你的风格的部分：

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()-function 不仅应用于列标题下方的单元格，还应用于包括列标题在内的整个列？

编辑： 为了澄清，这里是excel输出的截图： screenshot of excel output

“标题 2”应该与它下面的单元格具有相同的背景颜色。

Answer 1

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

使用@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文件的输出：

Answer 2

好的，我想我想出了一种方法来处理使用 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。