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

numpyserver side programmingprogramming更新于 2025/9/4 2:22:17

要返回一个索引数组,该数组沿指定轴对数组进行排序,请使用 Numpy 中的 ma​​.MaskedArray.argsort() 方法。轴使用"axis"参数设置,即排序的轴。

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

步骤

首先,导入所需的库 −

import numpy as np
import numpy.ma as ma

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

arr = np.array([[49, 85, 45], [67, 33, 59]])
print("数组...
", arr) print("
数组类型...
", arr.dtype)

获取数组的维度 −

print("数组维度...
",arr.ndim)

创建一个掩码数组并将其中一些掩码为无效 −

maskArr = ma.masked_array(arr, mask =[[0, 0, 1], [ 0, 1, 0]])
print("
我们的掩码数组
", maskArr) print("
我们的掩码数组类型...
", maskArr.dtype)

获取掩码数组的维度 −

print("
我们的掩码数组维度...
",maskArr.ndim)

获取掩码数组的维度 −

print("
我们的掩码数组形状...
",maskArr.shape)

获取掩码数组元素的数量 −

print("
掩码数组中的元素...
",maskArr.size)

返回一个包含索引的 ndarray,用于对数组进行排序,请使用 Numpy 中的 ma.MaskedArray.argsort() 方法。轴使用"axis"参数设置,即排序的轴。如果为 None(默认值),则使用扁平化数组。 −

print("
结果...
",maskArr.argsort(axis = None))

示例

import numpy as np
import numpy.ma as ma

# 使用 numpy.array() 方法创建一个包含 int 元素的数组
arr = np.array([[55, 85], [67, 33], [29, 88], [56, 45]])
print("数组...
", arr) print("
数组类型...
", arr.dtype) # 获取数组的维度 print("
数组维度...
",arr.ndim) # 创建一个掩码数组,并将其中一些掩码为无效 maskArr = ma.masked_array(arr, mask =[[1, 0], [ 0, 0], [0, 0], [0, 1]]) print("
我们的掩码数组
", maskArr) print("
我们的掩码数组类型...
", maskArr.dtype) # 获取掩码数组的维度 print("
我们的掩码数组维度...
",maskArr.ndim) # 获取掩码数组的形状 print("
我们的掩码数组形状...
",maskArr.shape) # 获取掩码数组的元素数量 print("
掩码数组中的元素...
",maskArr.size) # 要返回一个包含对数组进行排序的索引的 ndarray,请使用 Numpy 中的 ma.MaskedArray.argsort() 方法。 # 使用"axis"参数设置轴,即排序的轴。 # 如果为 None(默认值),则使用扁平化数组。 print("
结果...
",maskArr.argsort(axis = None))

输出

数组...
[[55 85]
[67 33]
[29 88]
[56 45]]

数组类型...
int64

数组维度...
2

我们的掩码数组
[[-- 85]
[67 33]
[29 88]
[56 --]]

我们的掩码数组类型...
int64

我们的掩码数组维度...

我们的掩码数组形状...
(4, 2)

掩码数组中的元素...
8

结果...
[4 3 6 2 1 5 0 7]

相关文章