如何在 Matplotlib imshow 中对矩阵应用掩码?

matplotlibpythondata visualization更新于 2025/9/12 5:00:17

要在 matplotlib imshow() 中对矩阵应用掩码，我们可以使用具有下限和上限的 np.ma.masked_where() 方法。

步骤

初始化两个变量 l 和 u，以掩码输入矩阵。
创建 5×5 维的随机数据。
掩码输入矩阵，l 值的较低值和 u 值的较高值。
创建一个图形和一组子图，其中 nrows=1 和 ncols=
将数据显示为图像，即在 2D 常规栅格上，轴 0 和
设置轴的标题，0和
要显示图形，请使用 show() 方法。

示例

import numpy as np
import matplotlib.pyplot as plt

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

l = 0.125
u = 0.575

data = np.random.rand(5, 5)
data = np.ma.masked_where((l < data) & (data < u), data)

fig, axs = plt.subplots(1, 2)

axs[0].imshow(data.data)
axs[0].set_title("Without Masked")
axs[1].imshow(data)
axs[1].set_title("With Masked")

plt.show()