蒙版3D numpy数组,其中array等于值列表

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

解决方法:

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,后者将生成一个掩码值被掩码的掩码数组.

然后,您可以对非屏蔽值运行操作,然后取消屏蔽,然后再对它们进行屏蔽或将其填充为默认值.

标签：masking,python,numpy
来源： https://codeday.me/bug/20191028/1953798.html