返回一個 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 次瀏覽

啟動您的職業生涯

完成課程獲得認證

開始學習
廣告