在 NumPy 中将掩码数组的 pickle 以字符串形式返回

numpyserver side programmingprogramming更新于 2025/9/26 21:07:17

要对掩码数组进行 pickle,请使用 ma.MaskedArray.dumps() 方法。使用 Numpy 中的 pickle.loads() 方法将 pickle 加载回数组。掩码数组是标准 numpy.ndarray 和掩码的组合。掩码可以是 nomask(表示关联数组中的任何值都有效),也可以是一个布尔值数组,用于确定关联数组中每个元素的值是否有效。

NumPy 提供全面的数学函数、随机数生成器、线性代数例程、傅里叶变换等。它支持各种硬件和计算平台,并且能够与分布式、GPU 和稀疏数组库良好兼容。

步骤

首先,导入所需的库 −

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)

要对掩码后的数组进行 pickle 操作,请使用 ma.MaskedArray.dumps() 方法 −

pickArr = maskArr.dumps()

显示 pickle 结果 −

print("
对掩码后的数组进行 pickle 操作...
",pickArr)

使用 pickle.loads() 方法将 pickle 加载回数组 −

resArr = pickle.loads(pickArr)

显示结果数组 −

print("
正在将 pickle 加载回掩码后的数组...
",resArr)

示例

import numpy as np
import numpy.ma as ma
import pickle

# 使用 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) # 要对掩码数组进行 pickle 操作,请使用 ma.MaskedArray.dumps() 方法 pickArr = maskArr.dumps() # 显示 pickle 值 print("
对掩码数组进行 pickle 操作...
",pickArr) # 使用 pickle.loads() 方法将 pickle 值加载回数组 resArr = pickle.loads(pickArr) # 显示结果数组 print("
正在将 pickle 值加载回掩码数组...
",resArr)

输出

数组...
[[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

Pickle of the masked array...
b"\x80\x02cnumpy.ma.core
_mareconstruct
q\x00(cnumpy.ma.core
MaskedArray
q\x01cnumpy
ndarray
q\x02K\x00\x85q\x03X\x01\x00\x00\x00bq\x04tq\x05Rq\x06(K\x01K\x04K\x04\x86q\x07cnumpy
dtype
q\x08X\x02\x00\x00\x00i8q\t\x89\x88\x87q
Rq\x0b(K\x03X\x01\x00\x00\x00<q\x0cNNNJ\xff\xff\xff\xffJ\xff\xff\xff\xffK\x00tq\rb\x89c_codecs
encode
q\x0eX\x80\x00\x00\x007\x00\x00\x00\x00\x00\x00\x00U\x00\x00\x00\x00\x00\x00\x00D\x00\x00\x00\x00\x00\x00\x00T\x00\x00\x00\x00\x00\x00\x00C\x00\x00\x00\x00\x00\x00\x00!\x00\x00\x00\x00\x00\x00\x00'\x00\x00\x00\x00\x00\x00\x005\x00\x00\x00\x00\x00\x00\x00\x1d\x00\x00\x00\x00\x00\x00\x00X\x00\x00\x00\x00\x00\x00\x003\x00\x00\x00\x00\x00\x00\x00%\x00\x00\x00\x00\x00\x00\x008\x00\x00\x00\x00\x00\x00\x00-\x00\x00\x00\x00\x00\x00\x00c\x00\x00\x00\x00\x00\x00\x00U\x00\x00\x00\x00\x00\x00\x00q\x0fX\x06\x00\x00\x00latin1q\x10\x86q\x11Rq\x12h\x0eX\x10\x00\x00\x00\x01\x01\x00\x00\x00\x00\x01\x00\x00\x00\x00\x01\x00\x01\x00\x00q\x13h\x10\x86q\x14Rq\x15Ntq\x16b." Loading the pickle back to the masked array... [[-- -- 68 84] [67 33 -- 53] [29 88 51 --] [56 -- 99 85]]

相关文章