在 Matplotlib 中仅绘制热图的上/下三角
matplotlibpythondata visualization
要在 matplotlib 中仅绘制热图的上/下三角，我们可以使用 numpy 获取掩码后的二维数组，然后将其转换为图像以生成热图。
步骤
设置图形大小并调整子图之间和周围的填充。
创建 5×5 维的随机数据。
使用 numpy.tri() 方法创建一个数组，在给定对角线及其下方为 1，在其他地方为 0。
使用掩码数组获取掩码后的二维数组数据(使用步骤 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()
输出
