How to set the background color of a column in a matplotlib table?

To set the background color of a column 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()