如何在 Python 中保存数据框表？

Question

我有一个关于将数据框放入表格并将其保存为 png 文件的问题。

为此，我编写了如下所示的一些代码块。

df = pd.DataFrame({'count' : fdf.groupby(['year','Name']).size()}).reset_index()
df = df.sort_values(['year','count'], ascending=[True,False]).set_index(['year','Name'])
df = df.style.background_gradient(cmap='YlOrRd')
df

这是我的 df

              count
year    Name    
1950    a      3
        b      3
1951    c      3
        d      2
        e      1
...    ...    ...

然后我尝试使用如下所示的这段代码sn-p来保存结果，但没有成功。

plt.figure(figsize=[15, 15])

ax = plt.subplot()
ax.xaxis.set_visible(False)
ax.yaxis.set_visible(False)
table(ax,f1_df_win_season.data)
plt.savefig('images/image1.png')

虽然我可以在 JupyterNotebook 的表格中看到所有变量，但在 png 文件中却看不到。

我该如何解决？

这是我的截图

Answer 1

我建议使用 matplotlib 'table' 函数创建一个表格，然后将所需的格式设置为背景渐变。

创建数据框后

df = pd.DataFrame({'count' : fdf.groupby(['year','Name']).size()}).reset_index()
df = df.sort_values(['year','count'], ascending=[True,False]).set_index(['year','Name'])

您可以像这样创建（并保存）您的表格

plt.figure(figsize=[15, 15])

ax = plt.subplot()
ax.xaxis.set_visible(False)
ax.yaxis.set_visible(False)
ax.table(cellText=df.reset_index().values,
        colLabels=df.reset_index().columns,
        loc='center',
        cellLoc='center')
plt.savefig('image1.png')

您可以在表格函数中添加所有格式（见https://matplotlib.org/3.1.1/api/_as_gen/matplotlib.pyplot.table.html）