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

Python中的load_npz()函数解读:用法和使用示例

发布时间:2023-12-11 04:33:36

在Python中,load_npz()函数是numpy库中的一个函数,用于加载保存为.npz文件格式的数据。

**用法:**

numpy.load_npz(file)

其中,file参数是要加载的.npz文件的路径。

**使用示例:**

假设我们有一个.npz文件,保存了两个数组arr1arr2,我们可以通过load_npz()函数来加载这些数据。

首先,让我们创建一个.npz文件并保存一些数据:

import numpy as np

arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([6, 7, 8, 9, 10])

np.savez('data.npz', arr1=arr1, arr2=arr2)

上述代码将创建一个名为data.npz的文件,并将arr1arr2保存为.npz文件中的两个数组。

现在,我们可以使用load_npz()函数来加载这些数据:

import numpy as np

data = np.load_npz('data.npz')

arr1_loaded = data['arr1']
arr2_loaded = data['arr2']

print(arr1_loaded)  # 输出: [1 2 3 4 5]
print(arr2_loaded)  # 输出: [ 6  7  8  9 10]

上述代码首先通过load_npz()函数加载了data.npz文件,并将返回的data对象保存在变量中。

然后,我们可以通过使用数组名称的方式从data对象中获取相应的数组。例如,arr1可以通过data['arr1']来获取。

最后,我们打印了从.npz文件中加载的这两个数组,可以看到成功加载并打印了这些数据。

总结来说,load_npz()函数是用于加载保存为.npz文件格式的数据的一个函数。它可以帮助我们从存储的.npz文件中读取数据,并将其存储为numpy数组供后续处理和分析使用。