np.save 函数 - Python

简介

np.save 是 NumPy 库中的一个函数，用于将数组数据保存到磁盘文件，以便以后重新加载使用。它方便地将大型数组保存到磁盘，避免了每次程序运行时重新生成数据。

使用方法

np.save 函数的语法如下：

numpy.save(file, arr, allow_pickle=True, fix_imports=True)

参数解释：

• file：需要保存数据的文件名或文件句柄。如果文件后缀没有指定，它将自动以.npy扩展名保存文件。
• arr：要保存的数组数据。
• allow_pickle：可选参数，设置为True允许以.pickle文件格式存储对象数组。默认为True。
• fix_imports：可选参数，用于Python2中导入Python3保存的pickle文件。默认为True。

示例

import numpy as np

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

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

# 重新加载数据
loaded_arr = np.load('array.npy')

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

这将输出：

[1 2 3 4 5]

使用pickle保存对象数组

import numpy as np

# 创建一个对象数组
arr = np.array([{'name': 'Alice', 'age': 25}, {'name': 'Bob', 'age': 30}])

# 保存对象数组到文件
np.save('objects.npy', arr)

# 重新加载对象数组
loaded_arr = np.load('objects.npy', allow_pickle=True)

# 打印加载的对象数组
for obj in loaded_arr:
    print(obj)

这将输出：

{'name': 'Alice', 'age': 25}
{'name': 'Bob', 'age': 30}

总结

通过 np.save 函数，我们可以方便地将数组数据保存到磁盘文件，并在需要时重新加载使用。这对于处理大型数据集或保存和共享模型参数等任务非常有用。