How to assign specific colors to specific cells in a Matplotlib table?

MatplotlibPythonData Visualization

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

raja
Published on 15-Jun-2021 12:27:10
Advertisements