导出pandas样式化表格到图像文件

Question

在jupyter笔记本中运行时,下面的代码呈现一个颜色渐变格式的表,我想将其导出到图像文件.

笔记本呈现的结果"styled_table"对象是pandas.io.formats.style.Styler类型.

我一直无法找到将Styler导出到图像的方法.

我希望有人可以分享一个出口的工作示例,或者给我一些指示.

import pandas as pd
import seaborn as sns

data = {('count', 's25'): 
       {('2017-08-11', 'Friday'): 88.0,
        ('2017-08-12', 'Saturday'): 90.0,
        ('2017-08-13', 'Sunday'): 93.0},
        ('count', 's67'): 
       {('2017-08-11', 'Friday'): 404.0,
        ('2017-08-12', 'Saturday'): 413.0,
        ('2017-08-13', 'Sunday'): 422.0},
        ('count', 's74'): 
       {('2017-08-11', 'Friday'): 203.0,
        ('2017-08-12', 'Saturday'): 227.0,
        ('2017-08-13', 'Sunday'): 265.0},
        ('count', 's79'): 
       {('2017-08-11', 'Friday'): 53.0,
        ('2017-08-12', 'Saturday'): 53.0,
        ('2017-08-13', 'Sunday'): 53.0}}

table = pd.DataFrame.from_dict(data)
table.sort_index(ascending=False, inplace=True)

cm = sns.light_palette("seagreen", as_cmap=True)
styled_table = table.style.background_gradient(cmap=cm)
styled_table

Answer 1

如评论中所述,您可以使用该render属性获取样式表的HTML:

html = styled_table.render()

然后,您可以使用将html转换为图像的包.例如,IMGKit:HTML到IMG包装器的Python库.请记住,此解决方案需要安装wkhtmltopdf,这是一种命令行工具,可将HTML呈现为PDF和各种图像格式.这一切都在IMGKit页面中描述.

一旦你有了,其余的很简单:

import imgkit
imgkit.from_string(html, 'styled_table.png')

Answer 2

您可以使用dexplo的dataframe_image从https://github.com/dexplo/dataframe_image。安装包后，它还允许您将样式器对象保存为图像，如本例中的图像README：

import numpy as np
import pandas as pd
import dataframe_image as dfi

df = pd.DataFrame(np.random.rand(6,4))
df_styled = df.style.background_gradient()

dfi.export(df_styled, 'df_styled.png')