在 NumPy 中返回沿軸 1 對掩碼陣列進行排序的索引的 ndarray

要返回對陣列進行排序的索引的 ndarray，請在 Numpy 中使用 **ma.MaskedArray.argsort()** 方法。軸使用“axis”引數設定，即要排序的軸。

返回沿指定軸對 a 進行排序的索引陣列。換句話說，a[index_array] 會產生排序後的 a。axis 是要排序的軸。如果為 None（預設值），則使用扁平化陣列。order 是當 a 是一個具有定義欄位的陣列時，此引數指定首先比較哪些欄位，其次比較哪些欄位，依此類推。不需要指定所有欄位。fill_value 是內部用於掩碼值的 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”引數設定，即要排序的軸。此處，axis 設定為值 1 -

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

示例

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.
# Here, axis is set with value 1
print("
Result...
",maskArr.argsort(axis = 1))

輸出

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...
[[1 0]
[1 0]
[0 1]
[0 1]]

AmitDiwan

更新於： 2022年2月4日

111 次檢視

啟動您的 職業生涯

透過完成課程獲得認證

開始學習