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])