Python - 屏蔽網格的特定值

Question

我想屏蔽網格的值。 例如，我想掩蓋「t < 0」的所有值以進行計算。 我嘗試使用conditionnal如果，但它不工作...

import numpy as np 

Lx=10. 
Ly=10. 

x0 = 2 

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

t = XA - 2 

Answer 1

你需要解釋一下你想要做後你掩蓋陣列什麼。你想改變未被屏蔽的值嗎？然後

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

可能是你所需要的。

在另一方面，如果你需要計算與屏蔽值數組的統計數據，你會發現使用NumPy masked arrays更方便：

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