返回非零未掩码元素的索引，并在 NumPy 中按元素对索引进行分组
numpyserver side programmingprogramming更新于 2025/9/26 14:52:17
要返回非零未掩码元素的索引，请使用 Numpy 中的 ma.MaskedArray.nonzero() 方法。要按元素对索引进行分组，我们在 transpose() 函数中使用了 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("数组...
", arr)
print("
数组类型...
", arr.dtype)
获取数组的维度 −
print("数组维度...
",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("
我们的掩码数组
", maskArr)
print("
我们的掩码数组类型...
", maskArr.dtype)
获取掩码数组的维度 −
print("
我们的掩码数组维度...
",maskArr.ndim)
获取掩码数组的维度 −
print("
我们的掩码数组形状...
",maskArr.shape)
获取掩码数组元素的数量 −
print("
掩码数组中的元素...
",maskArr.size)
要返回非零未掩码元素的索引，请使用 Numpy 中的 ma.MaskedArray.nonzero() 方法。该方法返回一个数组元组，每个维度一个，包含该维度中非零元素的索引。为了按元素对索引进行分组，我们在 transpose() 函数中使用了 nonzero() 方法。 −
print("
结果...
",np.transpose(maskArr.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("数组...
", arr)
print("
数组类型...
", arr.dtype)
# 获取数组的维度
print("
数组维度...
",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("
我们的掩码数组
", maskArr)
print("
我们的掩码数组类型...
", maskArr.dtype)
# 获取掩码数组的维度
print("
我们的掩码数组维度...
",maskArr.ndim)
# 获取掩码数组的形状
print("
我们的掩码数组形状...
",maskArr.shape)
# 获取掩码数组的元素数量
print("
掩码数组中的元素...
",maskArr.size)
# 要返回非零未掩码元素的索引，请使用 Numpy 中的 ma.MaskedArray.nonzero() 方法。
# 返回一个数组元组，每个维度一个，包含该维度中非零元素的索引。
# 要按元素对索引进行分组，我们使用了 transpose() 函数中的 nonzero() 方法。
print("
结果...
",np.transpose(maskArr.nonzero()))
输出
数组... [[55 85 59 77] [67 33 39 57] [29 88 51 37] [56 45 99 85]] 数组类型... int64 数组维度... 2 Our Masked Array [[-- -- 59 77] [67 33 -- 57] [29 88 51 --] [56 -- 99 85]] 我们的掩码数组类型... int64 我们的掩码数组维度... 我们的掩码数组形状... (4, 4) 掩码数组中的元素... 16 结果... [[0 2] [0 3] [1 0] [1 1] [1 3] [2 0] [2 1] [2 2] [3 0] [3 2] [3 3]]
