Python/Matplotlib:根据自定义/更改的色彩映射对表格中的单元格进行着色

import matplotlib.pyplot as plt
import matplotlib.colors as mcolors

# 定义颜色映射规则，例如使用灰度映射
cmap = mcolors.LinearSegmentedColormap.from_list("", ["white", "gray"])

# 创建表格
table_data = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

# 创建图表和子图
fig, ax = plt.subplots()

# 创建表格对象
table = ax.table(cellText=table_data, cellLoc='center', loc='center')

# 遍历表格中的每个单元格，设置背景色
for i, row in enumerate(table_data):
    for j, value in enumerate(row):
        cell = table.get_celld()[i, j]
        cell.set_facecolor(cmap(value))

# 隐藏表格边框
ax.axis('off')

# 显示图表
plt.show()