这是我的DataFrame:
import pandas as pd
import numpy as np
df = pd.DataFrame({'a': ['long', 'short', np.nan], 'b': [np.nan, 3, 2]})
我想设计df.to_excel
的输出.就像这样:
如果是df.a == 'long'
,a
是绿色,如果是df.a == 'short'
,a
是红色.
我已经读了很多关于SO的帖子,但我想不出该怎么做.
这是我的一次try ,非常接近结果:
def bounded_highlights(df):
conds = [df.a == 'short', df.a == 'long']
labels = ['background-color: pink', 'background-color: lime']
array = np.select(conds, labels, default='')
return array
当我使用df.style.apply(bounded_highlights, axis=None).to_excel(...
的时候,它不起作用.