[英]Python Pandas: Style column header
我正在使用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”应具有与其下方单元格相同的背景颜色。
好的,我想我想出了一种方法来处理使用 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。
您可以使用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)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.