python - 如何将 `style` 与 DataFrame 上的 `to_html` 类结合使用？

Question

我有一个像

这样的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 中有一个用于 float 的样式器会很好。喜欢

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 类/格式化程序？

Answer 1

试试这个:

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)

结果: