在 Matplotlib/Seaborn 绘图中为特定单元格添加自定义边框

在 Matplotlib/Seaborn 绘图中为特定单元格添加自定义边框。

步骤

  • 设置图形大小并调整子图之间的填充和子图周围的填充。
  • 使用某些列创建数据框。
  • 将矩阵数据集绘制为分层聚类的热图。
  • 获取热图轴作为子图排列。
  • 若要在 Matplotlib 中为特定单元格添加自定义边框，我们可以初始化一个变量border_color。
  • 使用自定义边框颜色，在热图轴上添加一个矩形贴片。
  • 若要显示图形，请使用show()方法。

示例

import pandas as pd
from matplotlib import pyplot as plt
import seaborn as sns
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
df = pd.DataFrame({"col1": [1, 4, 2, 3, 5], "col2": [3, 4, 1, 5, 2]})
g = sns.clustermap(df, figsize=(7.50, 3.50))
ax = g.ax_heatmap
border_color = "yellow"
ax.add_patch(plt.Rectangle((1, 2), 2, 1, fill=False,
edgecolor=border_color, lw=5))
plt.show()

输出

更新日期：2021 年 6 月 1 日

832 次浏览

开启你的职业

通过完成课程获得认证

开始
广告