Python numpy ma.notmasked_edges用法及代码示例

本文简要介绍 python 语言中 numpy.ma.notmasked_edges 的用法。

用法:

ma.notmasked_edges(a, axis=None)

沿轴查找第一个和最后一个未屏蔽值的索引。

如果所有值都被屏蔽，则返回 None。否则，返回两个元组的列表，分别对应第一个和最后一个未屏蔽值的索引。

参数：

a： array_like

输入数组。

axis： 整数，可选

沿其执行操作的轴。如果 None (默认)，适用于数组的扁平化版本。

返回：

edges： ndarray 或列表

如果数组中有任何掩码数据，则为开始和结束索引数组。如果数组中没有掩码数据，则边是第一个和最后一个索引的列表。

例子：

>>> a = np.arange(9).reshape((3, 3))
>>> m = np.zeros_like(a)
>>> m[1:, 1:] = 1

>>> am = np.ma.array(a, mask=m)
>>> np.array(am[~am.mask])
array([0, 1, 2, 3, 6])

>>> np.ma.notmasked_edges(am)
array([0, 6])