欢迎访问宙启技术站
智能推送

使用numpy.lib.format保存和加载NumPy数组

发布时间:2024-01-14 03:22:26

通过使用numpy.lib.format模块,我们可以保存和加载NumPy数组。这个模块提供了一些函数,可以将数组保存到磁盘上的二进制文件,并从文件中加载数组。

首先,我们需要导入numpynumpy.lib.format模块:

import numpy as np
import numpy.lib.format as fmt

接下来,我们可以使用fmt.write_array函数保存数组。该函数的参数包括一个文件对象和一个NumPy数组。以下是一个保存数组的示例:

arr = np.array([1, 2, 3, 4, 5])
file = open("array.bin", "wb")    # 打开一个二进制文件
fmt.write_array(file, arr)        # 保存数组到文件
file.close()                      # 关闭文件

在这个示例中,我们创建了一个包含整数1到5的NumPy数组。然后,我们打开一个名为"array.bin"的二进制文件,并使用fmt.write_array函数将数组保存到文件中。最后,我们关闭文件。

要加载保存的NumPy数组,我们可以使用fmt.read_array函数。以下是一个加载数组的示例:

file = open("array.bin", "rb")    # 打开二进制文件
arr_loaded = fmt.read_array(file) # 从文件中加载数组
file.close()                      # 关闭文件

print(arr_loaded)                 # 打印加载的数组

在这个示例中,我们打开之前保存的"array.bin"文件,并使用fmt.read_array函数从文件中加载数组。最后,我们关闭文件并打印加载的数组。

这是一个保存和加载多维数组的示例:

arr = np.array([[1, 2, 3], [4, 5, 6]])
file = open("array.bin", "wb")    # 打开一个二进制文件
fmt.write_array(file, arr)        # 保存数组到文件
file.close()                      # 关闭文件

file = open("array.bin", "rb")    # 打开二进制文件
arr_loaded = fmt.read_array(file) # 从文件中加载数组
file.close()                      # 关闭文件

print(arr_loaded)                 # 打印加载的数组

在这个示例中,我们创建了一个包含两行三列的NumPy数组。我们通过与之前相同的方式将数组保存到文件中,并使用相同的方式从文件中加载数组。最后,我们打印加载的数组。

使用numpy.lib.format模块可以很方便地保存和加载NumPy数组。我们可以通过fmt.write_array函数将数组保存到文件中,并通过fmt.read_array函数从文件中加载数组。这些函数提供了对NumPy数组的持久化的操作,非常适用于保存和加载大型数据集。