在 NumPy 中返回一個由掩碼陣列中給定索引處的元素組成的陣列

要返回一個由掩碼陣列中給定索引處的元素組成的陣列，請在 Python NumPy 中使用 **ma.MaskedArray.take()** 方法。

take() 方法返回的陣列與陣列具有相同的型別。indices 引數是要提取的值的索引。axis 引數是要選擇值的軸。預設情況下，使用扁平化的輸入陣列。如果提供了 out 引數，則結果將放置在此陣列中。它應該具有適當的形狀和 dtype。請注意，如果 mode='raise'，則 out 始終被緩衝；使用其他模式以獲得更好的效能。

mode 引數指定超出範圍的索引的行為方式。

• ‘raise’ - 丟擲錯誤（預設）

• ‘wrap’ - 環繞

• ‘clip’ - 裁剪到範圍

‘clip’ 模式意味著所有過大的索引都將替換為沿該軸定址最後一個元素的索引。

步驟

首先，匯入所需的庫 -

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)

給定的索引，即要提取的值的索引 -

indices = [4, 5, 8, 13, 14]

要返回一個由掩碼陣列中給定索引處的元素組成的陣列，請使用 ma.MaskedArray.take() 方法 -

print("
Result...
",np.take(maskArr, indices))

示例

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, 68, 84], [67, 33, 39, 53], [29, 88, 51, 37], [56, 45, 99, 85]])
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, 1, 0, 0], [ 0, 0, 1, 0],
[0, 0, 0, 1], [0, 1, 0, 0]])
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)

# The given indices i.e. the indices of the values to extract
indices = [4, 5, 8, 13, 14]

# To return an array formed from the elements of a masked array at the given indices, use the take() method
print("
Result...
",np.take(maskArr, indices))

輸出

Array...
[[55 85 68 84]
[67 33 39 53]
[29 88 51 37]
[56 45 99 85]]

Array type...
int64

Array Dimensions...
2

Our Masked Array
[[-- -- 68 84]
[67 33 -- 53]
[29 88 51 --]
[56 -- 99 85]]

Our Masked Array type...
int64

Our Masked Array Dimensions...
2

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

Elements in the Masked Array...
16

Result...
[67 33 29 -- 99]

AmitDiwan

更新於: 2022-02-04

545 次瀏覽

開啟你的 職業生涯

透過完成課程獲得認證

開始學習