NumPy 文件输入输出

介绍

在本实验中，你将学习如何使用 NumPy 读取和写入数组到文件。NumPy 提供了多个用于文件输入和输出的函数，使得处理大型数据集变得非常容易。

学习目标

• savetxt() 函数
• save() 函数
• loadtxt() 函数
• genfromtxt() 函数
• load() 函数

将数组写入文件

NumPy 提供了多个函数用于将数组写入文件。最常用的是 savetxt 和 save。

打开 Python Shell

在终端中输入以下命令以打开 Python shell：

python3

导入 NumPy

NumPy 已经安装，你可以在 Python 代码中导入它：

import numpy as np

使用 Savetxt

savetxt 函数用于将数组写入文本文件。以下是一个示例：

data = np.random.rand(10, 5)
np.savetxt('data.txt', data, delimiter=',')

• 这会将 data 的内容写入名为 data.txt 的文本文件，并用 逗号 分隔值。

使用 save

save 函数用于将数组写入二进制文件。以下是一个示例：

np.save('data.npy', data)

• 这会将 data 的内容写入名为 data.npy 的二进制文件。

从文件中读取数组

NumPy 提供了多个函数用于从文件中读取数组。最常用的是 loadtxt、genfromtxt 和 load。

使用 loadtxt

loadtxt 函数用于从文本文件中读取数组。以下是一个示例：

data = np.loadtxt('data.txt',delimiter=',')
print(data)

• 这会将 data.txt 的内容读取到一个 NumPy 数组中。从步骤 1 中我们知道，data.txt 中的值是用 逗号 分隔的。
• 代码 print(data) 会打印从 data.txt 中读取的内容。

使用 Genfromtxt

genfromtxt 函数与 loadtxt 类似，但它可以处理缺失值和其他特殊情况。以下是一个示例：

data = np.genfromtxt('data.txt', delimiter=',')
print(data)

• 这会将 data.txt 的内容读取到一个 NumPy 数组中。

使用 Load

load 函数用于从二进制文件中读取数组。以下是一个示例：

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

• 这会将 data.npy 的内容读取到一个 NumPy 数组中。

总结

恭喜！你已经完成了 NumPy 文件输入输出实验。

在本实验中，你学习了如何：

• 使用 NumPy 的 savetxt() 和 save() 函数将数组写入文件。
• 使用 NumPy 的 loadtxt()、genfromtxt() 和 load() 函数从文件中读取数组。