Python numpy ma.argsort用法及代碼示例

本文簡要介紹 python 語言中 numpy.ma.argsort 的用法。

用法:

ma.argsort(a, axis=<no value>, kind=None, order=None, endwith=True, fill_value=None)

返回一個 ndarray 索引，該索引沿指定軸對數組進行排序。預先填充屏蔽值fill_value.

參數：

axis： 整數，可選

要排序的軸。如果 None 是默認值，則使用展平數組。

kind： {‘quicksort’, ‘mergesort’, ‘heapsort’, ‘stable’}，可選

使用的排序算法。

order： 列表，可選

當 a 是定義了字段的數組時，此參數指定首先比較哪些字段，第二個等。並非所有字段都需要指定。

endwith： {真，假}，可選

是否應將缺失值(如果有)視為最大值 (True) 或最小值 (False) 當數組包含數據類型相同極端的未屏蔽值時，這些值和屏蔽值的順序未定義。

fill_value： 標量或無，可選

內部用於掩碼值的值。如果 fill_value 不是 None，它將取代 endwith 。

返回：

index_array： ndarray，int

排序的索引數組a沿著指定的軸。換句話說，a[index_array]產生一個排序的a.

注意：

有關不同排序算法的說明，請參閱 sort 。

例子：

>>> a = np.ma.array([3,2,1], mask=[False, False, True])
>>> a
masked_array(data=[3, 2, --],
             mask=[False, False,  True],
       fill_value=999999)
>>> a.argsort()
array([1, 0, 2])