numpy savez详解|极客笔记

numpy savez详解

在Python中，numpy是一个非常流行的用于科学计算的库，它提供了许多高效的数学函数和数据结构，可以方便地进行数组操作、线性代数运算等。savez是numpy中的一个函数，用于将多个数组保存到一个压缩文件中。

本文将详细介绍numpy中的savez函数，包括其语法、参数、用法示例以及相关注意事项。

语法和参数

savez函数的语法如下：

numpy.savez(file, *args, **kwds)

其中，参数说明如下：

file：保存到的文件名，可以是字符串或者类文件对象。
*args：一个或多个数组，用逗号分隔。
**kwds：其他关键字参数，例如allow_pickle等。

用法示例

下面我们来看一个简单的示例，展示如何使用savez函数保存多个数组到一个压缩文件中。

import numpy as np

# 创建两个数组
arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([[1, 2, 3], [4, 5, 6]])

# 保存数组到文件中
np.savez('arrays.npz', arr1=arr1, arr2=arr2)

# 读取保存的文件
data = np.load('arrays.npz')
print(data.files)  # 打印保存的数组

# 获取保存的数组
arr1_loaded = data['arr1']
arr2_loaded = data['arr2']

print(arr1_loaded)
print(arr2_loaded)

在上面的示例中，我们创建了两个数组arr1和arr2，然后使用savez函数将它们保存到一个名为arrays.npz的压缩文件中。接着我们使用load函数读取保存的文件，并打印出保存的数组arr1和arr2。