如何在Matplotlib imshow中对矩阵应用掩码?
要在matplotlib imshow() 中对矩阵应用掩码,我们可以使用带有下限和上限的 np.ma.masked_where() 方法。
步骤
- 初始化两个变量l和u,以掩盖输入矩阵。
- 创建5×5维度的随机数据。
- 掩盖输入矩阵,低于l值和高于u值。
- 创建一个具有nrows = 1和ncols =的图形和一组子图
- 将数据显示为图像,即在2D常规光栅上,在轴0和
- 设置轴0和1的标题。
- 要显示图形,请使用 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()