在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()

输出