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

如何使用Python中的load_npz()函数加载保存的npz文件

发布时间:2023-12-11 04:28:58

在Python中,我们可以使用numpy库中的load_npz()函数来加载保存的npz文件。该函数用于从磁盘中加载包含数组的.npz文件,并将其返回为一个字典对象,其中包含已保存的数组。

下面是一个使用load_npz()函数加载npz文件的示例:

首先,我们需要安装numpy库。可以使用以下命令在命令行中安装该库:

pip install numpy

接下来,可以使用以下代码来加载并查看保存的npz文件中的数组:

import numpy as np

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

# 查看npz文件中存在的数组
print(data.files)

# 读取数组内容
arr1 = data['arr_0']
arr2 = data['arr_1']

# 打印数组内容
print('Array 1:')
print(arr1)

print('Array 2:')
print(arr2)

在这个例子中,我们首先使用np.load()函数加载名为'example.npz'的npz文件。然后,我们使用data.files查看npz文件中存在哪些数组。通过data['arr_0']和data['arr_1']分别访问并保存npz文件中的数组arr1和arr2。最后,我们打印这两个数组的内容。

需要注意的是,当使用np.load()函数加载npz文件时,返回的是一个类似字典的对象。可以使用data.files查看文件中包含的所有数组的键。使用data['键名']来访问特定的数组。

另外,npz文件可以包含多个数组,可以使用不同的键名来访问这些数组。在上述示例中,我们使用了arr_0和arr_1两个键名来访问保存的数组。

使用load_npz()函数,我们可以轻松地加载保存的npz文件,方便地读取和使用其中的数组数据。这对于处理大型数据集和在不同的Python程序之间共享数据非常有用。