返回 NumPy 中非零且未掩碼元素的索引


要返回非零且未掩碼元素的索引,請使用 **ma.MaskedArray.nonzero()**

返回一個元組陣列,每個維度一個,包含該維度中非零元素的索引。可以使用 - 獲取相應的非零值。

a[a.nonzero()]

要按元素而不是維度對索引進行分組,請改用 -

np.transpose(a.nonzero())

其結果始終是一個二維陣列,每行對應一個非零元素。

步驟

首先,匯入所需的庫 -

import numpy as np
import numpy.ma as ma

使用 numpy.array() 方法建立一個包含 int 元素的陣列 -

arr = np.array([[55, 85, 59, 77], [67, 33, 39, 57], [29, 88, 51, 37], [56, 45, 99, 85]])
print("Array...
", arr) print("
Array type...
", arr.dtype)

獲取陣列的維度 -

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

建立一個掩碼陣列並掩蓋其中一些無效元素 -

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)

獲取掩碼陣列的維度 -

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

獲取掩碼陣列的形狀 -

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

獲取掩碼陣列的元素數量 -

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

要返回非零且未掩碼元素的索引,請在 NumPy 中使用 ma.MaskedArray.nonzero() 方法。返回一個元組陣列,每個維度一個,包含該維度中非零元素的索引 -

print("
Result...
",maskArr.nonzero())

示例

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, 59, 77], [67, 33, 39, 57], [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) # To return the indices of unmasked elements that are not zero, use the ma.MaskedArray.nonzero() method in Numpy # Returns a tuple of arrays, one for each dimension, containing the indices of the non-zero elements in that dimension. print("
Result...
",maskArr.nonzero())

輸出

Array...
[[55 85 59 77]
[67 33 39 57]
[29 88 51 37]
[56 45 99 85]]

Array type...
int64

Array Dimensions...
2

Our Masked Array
[[-- -- 59 77]
[67 33 -- 57]
[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...
(array([0, 0, 1, 1, 1, 2, 2, 2, 3, 3, 3]), array([2, 3, 0, 1, 3, 0, 1, 2, 0, 2, 3]))

更新於:2022年2月5日

201 次瀏覽

啟動您的 職業生涯

完成課程獲得認證

開始學習
廣告