numpy save

介绍

在进行数据处理和科学计算时，numpy 是一个非常重要的库，它提供了大量的数组操作函数和工具。numpy 提供了一种将数组保存到磁盘和从磁盘加载数组的方法。本文将详细介绍 numpy.save() 函数及其用法。

numpy.save() 函数的定义

numpy.save() 函数是 numpy 库中用于将数组保存到磁盘的函数。它的语法如下：

numpy.save(file, arr, allow_pickle=True, fix_imports=True)

• file: 字符串，文件名或文件对象，表示保存数组的位置。
• arr: 数组或类似对象，表示要保存的数组。
• allow_pickle: 可选参数，布尔值，默认为 True。如果设置为 False，则禁用嵌套对象字典的存储。
• fix_imports: 可选参数，布尔值，默认为 True。如果设置为 True，则为Python 2所写的文件添加main 方法。

numpy.save() 函数的用法

可以使用 numpy.save() 函数将数组保存到磁盘。下面是一些常见的用法示例。

示例1: 保存一维数组

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
np.save('arr.npy', arr)

上述代码将一维数组 arr 保存在当前目录下的 arr.npy 文件中。

示例2: 保存多维数组

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
np.save('arr.npy', arr)

上述代码将多维数组 arr 保存在当前目录下的 arr.npy 文件中。

示例3: 保存不同类型的数组

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([1.1, 2.2, 3.3])
arr3 = np.array(['a', 'b', 'c'])
np.savez('arr.npz', arr1=arr1, arr2=arr2, arr3=arr3)

上述代码使用 np.savez() 函数将不同类型的数组保存在 arr.npz 文件中。

示例4: 保存压缩文件

import numpy as np

arr = np.array([1, 2, 3, 4, 5])
np.savez_compressed('arr.npz', arr=arr)

上述代码使用 np.savez_compressed() 函数将数组保存在压缩文件 arr.npz 中。

numpy.load() 函数的定义

numpy.load() 函数是 numpy 中用于从磁盘加载数组的函数。它的语法如下：

numpy.load(file, mmap_mode=None, allow_pickle=True, fix_imports=True, encoding='ASCII')

• file: 字符串，文件名或文件对象，表示要加载的文件的位置。
• mmap_mode: 可选参数，字符串，默认为 None。如果设置为 None，返回通常的数组。如果设置为 ‘r’，返回只读数组。如果设置为 ‘r+’，返回可读写数组。如果设置为 ‘w+’，为二进制文件返回一个全新的可读写的数组。如果设置为 ‘c’，返回一个可写的而且存放在内存映像文件中的数组。如果设置为 ‘r’ 其它非空字符串，则将文件映射为按需内存映像文件。
• allow_pickle: 可选参数，布尔值，默认为 True。如果设置为 False，则禁用嵌套对象字典的存储。
• fix_imports: 可选参数，布尔值，默认为 True。如果设置为 True，则为Python 2所写的文件添加main方法。
• encoding: 可选参数，字符串，默认为 ‘ASCII’。用于加载的文本数据的编码。

numpy.load() 函数的用法

可以使用 numpy.load() 函数从磁盘加载保存的数组。下面是一些常见的用法示例。

示例1: 加载保存的数组

import numpy as np

arr = np.load('arr.npy')
print(arr)

上述代码将 arr.npy 文件加载为数组并打印出来。

示例2: 加载保存的多个数组

import numpy as np

data = np.load('arr.npz')
arr1 = data['arr1']
arr2 = data['arr2']
arr3 = data['arr3']
print(arr1)
print(arr2)
print(arr3)

上述代码加载 arr.npz 中的多个数组，并分别打印出来。

总结

本文介绍了 numpy.save() 函数和 numpy.load() 函数的用法。numpy.save() 函数用于将数组保存到磁盘，而 numpy.load() 函数用于从磁盘加载保存的数组。读者可以根据自己的需要灵活运用这两个函数，以实现数据的保存和加载操作。