在 NumPy 中返回由给定索引处的掩码数组元素组成的数组
numpyserver side programmingprogramming更新于 2025/9/3 16:37:17
要返回由给定索引处的掩码数组元素组成的数组，请使用 Python Numpy 中的 ma.MaskedArray.take() 方法。
take() 方法返回的数组与数组类型相同。indices 参数是要提取值的索引。axis 参数是用于选择值的轴。默认情况下，使用扁平化的输入数组。如果提供了 out 参数，则结果将放置在此数组中。它应具有适当的形状和数据类型。请注意，如果 mode="raise"，则 out 始终处于缓冲状态;使用其他模式以获得更好的性能。
mode 参数指定超出范围的索引将如何表现。
"raise" - 引发错误(默认)
"wrap" - 环绕
"clip" - 裁剪至范围
"clip"模式意味着所有过大的索引都将被替换为该轴上最后一个元素的索引。
步骤
首先，导入所需的库 −
import numpy as np import numpy.ma as ma
使用 numpy.array() 方法创建一个包含 int 元素的数组 −
arr = np.array([[49, 85, 45], [67, 33, 59]])
print("数组...
", arr)
print("
数组类型...
", arr.dtype)
获取数组的维度 −
print("数组维度...
",arr.ndim)
创建一个掩码数组并将其中一些掩码为无效 −
maskArr = ma.masked_array(arr, mask =[[0, 0, 1], [ 0, 1, 0]])
print("
我们的掩码数组
", maskArr)
print("
我们的掩码数组类型...
", maskArr.dtype)
获取掩码数组的维度 −
print("
我们的掩码数组维度...
",maskArr.ndim)
获取掩码数组的维度 −
print("
我们的掩码数组形状...
",maskArr.shape)
获取掩码数组元素的数量 −
print("
掩码数组中的元素...
",maskArr.size)
给定的索引，即要提取的值的索引 −
indices = [4, 5, 8, 13, 14]
要返回由给定索引处的掩码数组元素组成的数组，请使用 ma.MaskedArray.take() 方法−
print("
结果...
",np.take(maskArr, indices))
示例
import numpy as np
import numpy.ma as ma
# 使用 numpy.array() 方法创建一个包含 int 元素的数组
arr = np.array([[55, 85, 68, 84], [67, 33, 39, 53], [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)
# 给定的索引，即要提取的值的索引
indices = [4, 5, 8, 13, 14]
# 要返回由给定索引处的掩码数组元素组成的数组，请使用 take() 方法
print("
结果...
",np.take(maskArr, indices))
输出
数组... [[55 85 68 84] [67 33 39 53] [29 88 51 37] [56 45 99 85]] 数组类型... int64 数组维度... 2 我们的掩码数组 [[-- -- 68 84] [67 33 -- 53] [29 88 51 --] [56 -- 99 85]] 我们的掩码数组类型... int64 我们的掩码数组维度... 我们的掩码数组形状... (4, 4) 掩码数组中的元素... 16 结果... [67 33 29 -- 99]
