如何为带颜色单元格的matplotlib表格添加底部边框

问题原因 #

设置cell.visible_edges = 'B'时，matplotlib会重置单元格的样式属性（包括背景色、字体颜色等），直接覆盖之前自定义的颜色配置。

修复方案 #

调整代码执行顺序：先统一设置所有单元格的底部边框，再重新应用自定义的颜色、对齐方式等样式；也可以在设置边框后，单独为需要变色的单元格恢复颜色设置。

以下是修改后的完整代码：

# 读取Excel文件
excel_file = "/path"
df = pd.read_excel(excel_file, sheet_name='Sheet1')

# 过滤全为空的行和列
df = df.dropna(axis=0, how='all')
df = df.dropna(axis=1, how='all')

# 创建matplotlib可视化表格
fig, ax = plt.subplots(figsize=(16, 10))
ax.axis('off')  # 隐藏坐标轴

# 生成表格，无表头、初始无边框
tab = ax.table(cellText=df.values, cellLoc='center', loc='center')

columns_to_adjust = [0, 1]  # 需要调整的列索引

# 第一步：先为所有单元格设置底部边框
for i in range(len(df)):
    for j in range(len(df.columns)):
        cell = tab[(i, j)]
        cell.visible_edges = 'B'

# 第二步：重新应用自定义样式和颜色设置
for i in range(len(df.iloc[:, 0])):
    for col_index in columns_to_adjust:
        cell = tab[(i, col_index)]
        cell_text = cell.get_text().get_text()
        cell.get_text().set_horizontalalignment('left')
        cell.set_height(0.04)
        
        # 调整第0列宽度和样式
        cell_col0 = tab[(i, 0)]
        cell_col0.set_width(0.45)
        cell_col0.PAD = 0.025
        if "ABC" in cell_text:
            cell_col0.get_text().set_color('white')
            cell_col0.set_facecolor('blue')
        
        # 调整第1列宽度
        cell_col1 = tab[(i, 1)]
        cell_col1.set_width(0.035)

plt.show()

关键说明 #

• 先设置边框再应用样式，避免边框设置覆盖之前的自定义配置
• 拆分原循环中的重复操作，减少重复获取单元格对象的冗余，避免样式冲突
• 如果仅需为特定列添加底部边框，可修改边框设置循环，只遍历目标列即可

内容的提问来源于stack exchange，提问作者MariusC