How can I save table of dataframe in Python?

Question

I have an issue about putting dataframe into to table and saving it as a png file.

To do that, I wrote some code blocks shown below.

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

Here is my df

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

Then I tried to use this code snippet shown below to save result but it didn't work.

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')

Although I can see all variables in table in JupyterNotebook, I couldn't see it in png file.

How can I fix it?

Here is my screenshot

Rik Kraan · Accepted Answer · 2020-07-06 18:14:59Z

1

I would recommend to create a table using the matplotlib 'table' function and do the desired formatting as background gradients afterwards.

After creating the dataframe with

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

you can create (and save) your table like

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')

You can add all the formatting in the table function (see https://matplotlib.org/3.1.1/api/_as_gen/matplotlib.pyplot.table.html)

answered Jul 6, 2020 at 18:14

Rik Kraan

5862 silver badges17 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

Sercan Noyan Germiyanoğlu Over a year ago

@Thank you for your response. Because there are 366 values in dataframe , some of them cannot be shown in the image1.png file. How can I fix it?

Rik Kraan Over a year ago

I guess adding bbox_inches='tight' to plt.savefig will do the trick plt.savefig('image1.png', bbox_inches="tight")

Ynjxsjmh Over a year ago

@TonyBrand Your linked question is already solved.

Collectives™ on Stack Overflow

How can I save table of dataframe in Python?

1 Answer 1

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related