Assign Specific Colors to Cells in a Matplotlib Table

To assign specific colors to specific cells in a Matplotlib table, we can take the following steps −

• Set the figure size and adjust the padding between and around the subplots.
• Make a tuple for columns attribute.
• Make a list of lists, i.e., list of records.
• Make a list of lists, i.e., color of each cell.
• Create a figure and a set of subplots.
• Add a table to an axes ax.
• Turn off the axes.
• To display the figure, use show() method.

Example

import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

columns = ('name', 'age', 'marks', 'salary')

cell_text = [["John", "23", "98", "234"],
["James", "24", "90", "239"]]

colors = [["red", "yellow", "blue", "green"],
["blue", "green", "yellow", "red"]]

fig, ax = plt.subplots()

the_table = ax.table(cellText=cell_text, cellColours=colors,
                     colLabels=columns, loc='center')
ax.axis('off')

plt.show()

Output