python - 屏蔽 3D numpy 数组，其中数组等于值列表

Question

如何使用整数列表屏蔽 3D numpy 数组？我希望数组中元素等于列表中任何值的所有元素都被屏蔽。

Answer 1

import numpy as np
import numpy.ma as ma

randomArray = np.random.random_integers(0, 10, (5, 5, 5))
maskingValues = [1, 2, 5]  
maskedRandomArray = ma.MaskedArray(randomArray, np.in1d(randomArray, maskingValues))

为了便于说明，上面的代码将创建一个 3D 数组，其中包含 0 到 10 之间的随机整数值。然后我们将定义要从第一个数组中屏蔽的值。然后我们使用 np.in1d 方法根据我们的原始数组和值创建一个 bool 掩码，并将其传递到 numpy.ma.MaskedArray 中。 ，它生成一个屏蔽数组，其中的值被屏蔽掉。

然后，这允许您对非屏蔽值运行操作，然后再取消屏蔽或使用默认值填充它们。