【Python】np.save()和np.load()函数详解和示例

本文通过函数原理和运行示例，对np.save()和np.load()函数进行详解，以帮助大家理解和使用。
更多Numpy函数详解和示例，可参考

【Python】Numpy库近50个常用函数详解和示例，可作为工具手册使用

np.save （）

np.save() 是 NumPy 库中的一个函数，用于将数组保存到 .npy 文件中。这种文件格式可以用于在之后重新加载数组。

函数解析

函数原型：
numpy.save(file, arr, allow_pickle=True, fix_imports=True, do_compression=False)

参数：
file：要保存到的 .npy 文件的路径。
arr：要保存的数组。
allow_pickle：布尔值，决定是否允许 pickling 数据。默认为 True。
fix_imports：布尔值，决定是否应该尝试修复不兼容的导入。默认为 True。
do_compression：布尔值，决定是否应该对文件进行压缩。默认为 False。

运行示例

import numpy as np

# 创建一个数组
arr = np.array([[1, 2], [3, 4]])

# 将数组保存到 .npy 文件
np.save('my_array.npy', arr)

同级别文件夹中生成了一个“my_array.npy”文件，内容为：

用记事本打开有些乱码。

np.load （）

np.load 是 NumPy 库中的一个函数，用于从 .npy 文件中加载数组。这种文件是由 np.save 函数创建的。np.load 函数非常适合用于恢复之前保存的数组，以便可以在之后的代码中使用。

函数解析

函数原型：
numpy.load(file, mmap_mode=None, allow_pickle=True, fix_imports=True, encoding='ASCII')

参数：
file：要加载的 .npy 文件的路径。
mmap_mode：可选参数，用于指定内存映射模式的文件加载。默认值是 None。
allow_pickle：布尔值，决定是否允许加载 pickle 数据。默认为 True。
fix_imports：布尔值，决定是否应该尝试修复不兼容的导入。默认为 True。
encoding：字符串，用于指定编码方式。默认为 'ASCII'。

运行示例

import numpy as np

# 创建一个数组
arr = np.array([[1, 2], [3, 4]])

# 将数组保存到 .npy 文件
np.save('my_array.npy', arr)
# 加载 .npy 文件
loaded_arr = np.load('my_array.npy')

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

输出：

[[1 2]
 [3 4]]