在 Matplotlib 中仅绘制热图的上/下三角

要在 matplotlib 中仅绘制热图的上/下三角，我们可以使用 numpy 来获取掩码的 2D 数组，并将它们转换为图像以生成热图。

步骤

  • 设置图形大小并调整子图之间和周围的填充。

  • 创建 5×5 维度的随机数据。

  • 使用 numpy.tri() 方法创建数组，其 1 值位于给定对角线及以下位置，在其他位置为 0。

  • 获取有掩码的 2D 数组数据（使用步骤 3）。

  • 使用 imshow() 方法将数据显示为图像，即显示在 2D 常规光栅上。

  • 要显示图形，请使用 show() 方法。

示例

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
data = np.random.rand(5, 5)
mask = np.tri(data.shape[0], k=-1)
data = np.ma.array(data, mask=mask)
plt.imshow(data, interpolation="nearest", cmap='copper')
plt.show()

输出

更新于： 01-Jun-2021

超过 1 千次浏览

开启您的 职业生涯

完成课程即可获得认证

开始
广告