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