Formatting Table in Pandas plot with table=true

Question

I have a dataframe that I have plotted as a stacked bar chart and added a table of values below the plot using table=true. I've had to modify it slightly so its readable.

myplot=df.plot.bar(stacked=True,title=name,grid=True,fontsize = 12, table=True)
myplot.axes.get_xaxis().set_visible(False)
table = myplot.tables[0]
# Setting the font size
table.set_fontsize(10)

Is it possible to:

Add another row to the table with the SUM values of each column
Remove NaNs
Add background color to the table

Thanks

Nathan Clement · Accepted Answer · 2019-11-27 17:09:56Z

1

Most of this information comes from digging around here:

I used the strategies recommended here: Matplotlib table formatting

For a complete solution, here is my setup (slight extension of yours):

import pandas as pd
import matplotlib.pyplot as plt

data = {
    1: {'A': 50, 'B': 100, 'C': 55, 'D': 42, 'E': 55},
    2: {'A': 100, 'B': 50, 'C': None, 'D': None, 'E': None},
    3: {'A': 67, 'B': None, 'C': None, 'D': None, 'E': None},
    4: {'A': 50, 'B': None, 'C': None, 'D': None, 'E': None}
}
df = pd.DataFrame(data)
#df = df.swapaxes("index", "columns")

myplot=df.plot.bar(stacked=True,grid=True,fontsize = 12, table=True)
myplot.axes.get_xaxis().set_visible(False)
table = myplot.tables[0]
# Setting the font size
table.set_fontsize(10)

This section iterates through cells in the table, checks if the text is "nan", sets to blank if it is, and sets the color:

# iterate through cells
desiredBgColor = (0.5,0.5,0.5)
tableRowLength = len(df.columns)
tableColLength = len(df.index)
for cIx in range(1, tableColLength): # start with 1 to not include column headers
    for rIx in range(0, tableRowLength + 1): # start with 0 because apparently row index does appear in celldata
        c = table.get_celld()[(cIx, rIx)]
        cellText = c.get_text()
        if (cellText.get_text() == 'nan'):
            cellText.set_text('')
        c.set_color(desiredBgColor)

This adds a sum row:

# add a row
desiredSumRowColor = (.6, .6, .6)
cellWidth = table.get_celld()[(1, 1)].get_width() # have to check width from an existing cell
cellHeight = table.get_celld()[(1, 1)].get_height() # have to check height from an existing cell
for cIx in range(0, tableColLength):
    c = table.add_cell(tableRowLength + 1, cIx, cellWidth, cellHeight)
    c.set_color(desiredSumRowColor)
    c.set
    c.get_text().set_text(df.iloc[cIx].sum())

Final result below:

answered Nov 27, 2019 at 17:09

Nathan Clement

1,2232 gold badges20 silver badges38 bronze badges

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

3 Comments

user11305439 Over a year ago

This is fantastic, thanks. Is it possible to get the grid lines back or is that asking too much?

Nathan Clement Over a year ago

@user11305439 c.set_edgecolor('black') will change the grid lines to black on a cell :)

Nathan Clement Over a year ago

Glad to help! If this solved your issue, could you accept the answer? (Green checkbox). Thanks :)

Collectives™ on Stack Overflow

Formatting Table in Pandas plot with table=true

1 Answer 1

3 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related