NumPy 输入输出

NumPy 提供用于加载和保存数据的输入/输出 (I/O) 函数。

输入/输出函数支持多种文件格式，包括二进制和文本格式。

• 二进制格式专为高效存储和检索大型数组而设计。
• 文本格式更易于人类阅读，并且可以轻松地在文本编辑器中进行编辑。

最常用的 I/O 函数

以下是一些常用的 NumPy 输入/输出函数

NumPy save() 函数

在 NumPy 中，save() 函数用于将数组以 NumPy .npy 格式保存到二进制文件。

以下是 save() 函数的语法，

np.save(file, array)

• file - 指定文件名 (如有需要，请包含路径)
• array - 指定要保存的 NumPy 数组

现在，让我们看一个示例。

import numpy as np

# create a NumPy array
array1 = np.array([[1, 3, 5], 
                   [7, 9, 11]])

# save the array to a file
np.save('file1.npy', array1)

在这里，我们将名为 array1 的 NumPy 数组保存到我们当前目录中名为 file1.npy 的二进制文件中。

NumPy load() 函数

在上一个示例中，我们将一个数组保存到了一个二进制文件。现在我们将使用 load() 函数加载该已保存的文件。

让我们看一个例子。

import numpy as np

# load the saved NumPy array
loaded_array = np.load('file1.npy')

# display the loaded array
print(loaded_array)

输出

[[ 1  3  5]
 [ 7  9 11]]

在这里，我们使用 load() 函数读取了名为 file1.npy 的二进制文件。这是我们在上一个示例中使用 save() 函数创建和保存的同一个文件。

NumPy savetxt() 函数

在 NumPy 中，我们使用 savetxt() 函数将数组保存到文本文件。

以下是 savetxt() 函数的语法

np.save(file, array)

• file - 指定文件名
• array - 指定要保存的 NumPy 数组

现在，让我们看一个示例，

import numpy as np

# create a NumPy array
array2 = np.array([[1, 3, 5], 
                   [7, 9, 11]])

# save the array to a file
np.savetxt('file2.txt', array2)

上面的代码会将名为 array2 的 NumPy 数组保存到我们当前目录中名为 file2.txt 的文本文件中。

NumPy loadtxt() 函数

我们使用 loadtxt() 函数加载已保存的 txt 文件。

让我们看一个加载我们之前保存的 file2.txt 文件的示例。

import numpy as np

# load the saved NumPy array
loaded_array = np.loadtxt('file2.txt')

# display the loaded array
print(loaded_array)

输出

[[1. 3. 5.]
 [7. 9. 11.]]

在这里，我们使用 loadtxt() 函数加载了之前使用 savetxt() 函数创建和保存的名为 file2.txt 的文本文件。

注意：加载的数组中的值带有小数点 .，因为 loadtxt() 默认将值读取为浮点数。