如何在DataFrame上结合使用`style`和`to_html`类?

时间:2017-03-06 15:16:31

标签: python pandas

我有一个像

这样的DataFrame
df = pd.DataFrame(np.random.randn(10).reshape(2, 5))

df
#              0         1         2         3         4
#    0 -0.067162 -0.505401 -0.019208  1.123936  0.087682
#    1 -0.373212 -0.598412  0.185211  0.736143 -0.469111

我正在尝试将此DataFrame输出为HTML,之前使用的是to_html

df.to_html(classes=['table', 'table-hover', 'table-bordered'], 
           float_format=lambda x: '{0:.3f}s'.format(x))

但后来我遇到了Style功能,并认为在我的DataFrame中为浮动设备设置样式会更好。像

def colorize(num)
    color = 'red' if (np.isnan(num) or num > 0) else 'green'
    return 'color: %s' % color

我可以使用

应用于我的DataFrame
df_styler = df.Style.applymap(colorize)

但现在df_styler是一个Styler对象,虽然它有render方法,但我不知道如何通过classes列表或我与to_html一起使用的浮动格式化程序......

我是否可以使用Style函数和to_html中的CSS类/格式化程序进行组合?

1 个答案:

答案 0 :(得分:8)

试试这个:

html = df.style.applymap(colorize) \
         .set_table_attributes('border="1" class="dataframe table table-hover table-bordered"') \
         .set_precision(3) \
         .render()

with open('d:/temp/a2.html', 'w') as f:
    f.write(html)

结果:

enter image description here