NumPy 函数

NumPy save() 函数用于将数组保存为 NumPy .npy 格式的二进制文件。

注意: 要从 .npy 文件加载数组,使用load()函数。

语法

numpy.save(file, arr) 

参数

file必填。 指定保存数据的文件或文件名。如果 file 是文件对象,则文件名不变。如果文件是字符串或路径,则将 .npy 扩展名附加到文件名(如果文件名尚无扩展名)。
arr必填。 指定要保存的数组数据。

返回值

无。

示例:

在下面的示例中,数组 arr 被保存到一个名为 test.npy 的新二进制文件中。此外,load()函数用于从文件中加载保存的数组并打印它

import numpy as np

arr = np.array([10, 20, 30, 40, 50, 60])

#将 arr 保存在二进制文件中 - test.npy
np.save("test", arr)

#从 test.npy 加载数组
y = np.load("test.npy")

#显示y的内容
print(y) 

上述代码的输出将是:

[10 20 30 40 50 60] 

示例:

假设我们有一个名为 demo.npy 的文件。下面的示例描述了如何在其中保存 numpy 数组并从中加载保存的数组。

import numpy as np

#以写入模式打开文件
#保存 numpy 数组
MyFile = open("demo.npy", "wb")
np.save(MyFile, np.array([10, 20]))
np.save(MyFile, np.array([10, 30]))
MyFile.close()

#打开文件读取内容
MyFile = open("demo.npy", "rb")
x = np.load(MyFile)
y = np.load(MyFile)
MyFile.close()

#显示x和y的内容
print(x)
print(y) 

上述代码的输出将是:

[10 20]
[10 30]