时间戳

首页 > 解决方案 > 来自 pandas DF 的表格 png 文件

python - 来自 pandas DF 的表格 png 文件

问题描述

大家早上好，我正在尝试将 pandas DF 导出为 png 文件。我正在使用来自 volodymyr 的代码，效果很好，但我也想获得行名。我试图修改一下代码，但我只能得到一个没有任何行名的空列。

任何建议。

def render_mpl_table(data, col_width=3.0, row_height=0.625, font_size=14,
                     header_color='#40466e', row_colors=['#f1f1f2', 'w'], edge_color='w',
                     bbox=[0, 0, 1, 1], header_columns=0,header_row=0,
                     ax=None, **kwargs):
    if ax is None:
        size = (np.array(data.shape[::-1]) + np.array([0, 1])) * np.array([col_width, row_height])
        fig, ax = plt.subplots(figsize=size)
        ax.axis('off')

    mpl_table = ax.table(cellText=data.values,colLabels=data.columns,rowLabels=data.index, bbox=bbox,**kwargs)

    mpl_table.auto_set_font_size(False)
    mpl_table.set_fontsize(font_size)

    for k, cell in  six.iteritems(mpl_table._cells):
        cell.set_edgecolor(edge_color)
        if k[0] == 0 or k[1] < header_columns:
            cell.set_text_props(weight='bold', color='w')
            cell.set_facecolor(header_color)
        else:
            cell.set_facecolor(row_colors[k[0]%len(row_colors) ])
    plt.tight_layout()
    plt.savefig("tabella.png")
    return ax

标签： pythonpandasdataframeplot

解决方案

只需使用 bbox_inches 紧密保存文件，现在我就有了包含列名和行名的所有表。

plt.savefig("tabella.png",bbox_inches="tight")

推荐阅读