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

Python中的load_npz()函数及其用法

发布时间:2023-12-11 04:27:59

load_npz()函数是Python中用于加载NumPy数组数据的函数。npz文件是NumPy保存数组数据的一种格式,它可以包含多个数组,并且每个数组都有一个名称。load_npz()函数可以用来读取npz文件中的数组数据,并将其返回为一个字典对象。

load_npz()函数的用法如下:

numpy.load_npz(file)

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

下面是一个使用load_npz()函数的例子:

import numpy as np

# 创建几个NumPy数组
arr1 = np.array([1, 2, 3])
arr2 = np.array([[4, 5, 6], [7, 8, 9]])

# 保存数组为npz文件
np.savez('data.npz', arr1=arr1, arr2=arr2)

# 加载npz文件
data = np.load('data.npz')

# 通过数组名称获取数组数据
loaded_arr1 = data['arr1']
loaded_arr2 = data['arr2']

# 打印加载的数组数据
print(loaded_arr1)
print(loaded_arr2)

运行以上代码,将得到以下输出:

[1 2 3]
[[4 5 6]
 [7 8 9]]

在上面的例子中,首先创建了两个NumPy数组arr1和arr2。然后,使用np.savez()函数将这两个数组保存为名为"data.npz"的npz文件。接下来,使用np.load()函数加载了"data.npz"文件,并将其返回的字典对象赋值给变量data。可以通过使用数组名称作为字典键来获取数组的数据。

load_npz()函数是一个非常方便的用于加载和读取NumPy数组数据的函数。而且,由于npz文件可以保存多个数组,所以可以在一个文件中同时保存和读取多个NumPy数组。