使用numpy.lib.format保存和加载NumPy数组
发布时间:2024-01-14 03:22:26
通过使用numpy.lib.format模块,我们可以保存和加载NumPy数组。这个模块提供了一些函数,可以将数组保存到磁盘上的二进制文件,并从文件中加载数组。
首先,我们需要导入numpy和numpy.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数组的持久化的操作,非常适用于保存和加载大型数据集。
