numpy.ma.make_mask()函数介绍

1. 简介

numpy.ma.make_mask() 是mask Module中的函数。

• numpy.ma是numpy中用于处理missing values的module。
• make_mask()用于生成一个新的mask object。

2. 函数定义

numpy.ma.make_mask(a, copy=True, shrink=True, dtype=None, order='K')

参数：

• a ：array_like, 要mask化的数组
• copy ：bool, 可选，是否复制源数组, 默认为 True
• shrink ：bool, 可选，是否将mask尽量的小, 默认为 True
• dtype ：data-type, 可选，定义生成mask数组的数据类型，默认为None，即默认数据类型
• order ：str, 可选，用于复制输入array数据的memory layout。表示内存块的排列方式，有8种选项，一般用默认方式'K'

返回值：

返回一个MaskedArray，是完全构建好的mask数组。

3. 使用示例

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
mask = np.ma.make_mask(arr==3)
# 上面的一行相当于下面两行的结合
# mask = np.ones_like(arr, bool)
# mask[np.where(arr==3)] = False

生成了一个与arr数组形状一致，数值全为True，除了等于3的数值对应的位置是False的mask数组。

返回值类型为MaskedArray:

print(mask) 
# mask result: [ True  True False  True  True]

print(type(mask)) 
# result: numpy.ma.core.MaskedArray

参考文献

• numpy.ma.make_mask()官方文档
• numpy.ma官方文档