返回一个 NumPy 中沿指定轴对掩码数组进行排序的索引 ndarray


要返回一个对数组进行排序的索引 ndarray,请在 NumPy 中使用 **ma.MaskedArray.argsort()** 方法。轴使用“axis”参数设置,即要排序的轴。

返回沿指定轴对 a 进行排序的索引数组。换句话说,a[index_array] 会产生一个已排序的 a。轴是要排序的轴。如果为 None(默认值),则使用扁平化数组。顺序是当 a 是一个定义了字段的数组时,此参数指定首先比较哪些字段,其次是哪些字段,依此类推。并非所有字段都需要指定。fill_value 是内部用于掩码值的数值。如果 fill_value 不为 None,则它会取代 endwith。

步骤

首先,导入所需的库:

import numpy as np
import numpy.ma as ma

使用 numpy.array() 方法创建一个包含整数元素的数组:

arr = np.array([[49, 85, 45], [67, 33, 59]])
print("Array...
", arr) print("
Array type...
", arr.dtype)

获取数组的维度:

print("Array Dimensions...
",arr.ndim)

创建一个掩码数组并掩盖其中一些无效值:

maskArr = ma.masked_array(arr, mask =[[0, 0, 1], [ 0, 1, 0]])
print("
Our Masked Array
", maskArr) print("
Our Masked Array type...
", maskArr.dtype)

获取掩码数组的维度:

print("
Our Masked Array Dimensions...
",maskArr.ndim)

获取掩码数组的形状:

print("
Our Masked Array Shape...
",maskArr.shape)

获取掩码数组的元素个数:

print("
Elements in the Masked Array...
",maskArr.size)

要返回一个对数组进行排序的索引 ndarray,请在 NumPy 中使用 ma.MaskedArray.argsort() 方法。轴使用“axis”参数设置,即要排序的轴。如果为 None(默认值),则使用扁平化数组:

print("
Result...
",maskArr.argsort(axis = None))

示例

import numpy as np
import numpy.ma as ma

# Create an array with int elements using the numpy.array() method
arr = np.array([[55, 85], [67, 33], [29, 88], [56, 45]])
print("Array...
", arr) print("
Array type...
", arr.dtype) # Get the dimensions of the Array print("
Array Dimensions...
",arr.ndim) # Create a masked array and mask some of them as invalid maskArr = ma.masked_array(arr, mask =[[1, 0], [ 0, 0], [0, 0], [0, 1]]) print("
Our Masked Array
", maskArr) print("
Our Masked Array type...
", maskArr.dtype) # Get the dimensions of the Masked Array print("
Our Masked Array Dimensions...
",maskArr.ndim) # Get the shape of the Masked Array print("
Our Masked Array Shape...
",maskArr.shape) # Get the number of elements of the Masked Array print("
Elements in the Masked Array...
",maskArr.size) # To return an ndarray of indices that sort the array, use the ma.MaskedArray.argsort() method in Numpy # The axis is set using the "axis" parameter i.e the Axis along which to sort. # If None, the default, the flattened array is used. print("
Result...
",maskArr.argsort(axis = None))

输出

Array...
[[55 85]
[67 33]
[29 88]
[56 45]]

Array type...
int64

Array Dimensions...
2

Our Masked Array
[[-- 85]
[67 33]
[29 88]
[56 --]]

Our Masked Array type...
int64

Our Masked Array Dimensions...
2

Our Masked Array Shape...
(4, 2)

Elements in the Masked Array...
8

Result...
[4 3 6 2 1 5 0 7]

更新于:2022年2月4日

105 次浏览

启动您的职业生涯

完成课程获得认证

开始学习
广告
© . All rights reserved.