如何在Matplotlib表格中设置列的背景色？

Matplotlib是一个Python数据可视化库，可以帮助我们创建各种图表和图形。Matplotlib还包含一个表格类，可以用于创建表格。如果你需要设置表格列的背景色，那么本文将会为你提供帮助。

创建表格

Matplotlib提供了table类来创建表格。下面是一个简单的示例代码，它将创建一个2×2的表格，并将它们打印出来。

import matplotlib.pyplot as plt

cells = [[1, 2], [3, 4]]
columns = ('A', 'B')
rows = ['1', '2']

fig, ax = plt.subplots(1,1)
table = ax.table(cellText=cells, colLabels=columns, rowLabels=rows, loc='center')
table.auto_set_font_size(False)
table.set_fontsize(14)
table.scale(1, 1.5)

plt.show()

设置列的背景色

如果你需要设置表格列的背景色，你需要将每列的背景色以列表形式传递给add_cell方法，然后添加到表格中。下面是一个示例代码：

import matplotlib.pyplot as plt

cells = [[1, 2], [3, 4]]
columns = ('A', 'B')
rows = ['1', '2']

fig, ax = plt.subplots(1,1)
table = ax.table(cellText=cells, colLabels=columns, rowLabels=rows, loc='center')
table.auto_set_font_size(False)
table.set_fontsize(14)
table.scale(1, 1.5)

# 设置表格列的背景色
colors = ['red', 'yellow']
for i in range(len(rows)):
    for j in range(len(columns)):
        if j == 1:
            table.get_celld()[(i,j)].set_facecolor(colors[i])

plt.show()

在此代码中，我们将每列的背景色作为列表colors传递给add_cell方法，然后使用get_celld方法获取单元格，并使用set_facecolor方法设置单元格的背景色。