Numpy Python-网格的掩码特定值

我想屏蔽网格的值。 例如，我想屏蔽“t<0”的所有值，以便在之后进行计算。 我尝试使用条件if，但它不起作用

import numpy as np

Lx=10.
Ly=10.

x0 = 2

YA, XA = np.mgrid[0:Ly, 0:Lx]

t = XA - 2

---

您需要解释在屏蔽阵列后要执行的操作。是否要更改未屏蔽的值？然后

mask = t < 0
YA[~mask] = ...

YA = np.ma.masked_array(YA, mask)