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数组。
