Numpy将对象数组写入.txt文件

在本文中，我们将介绍如何使用Numpy将对象数组写入.txt文件。Numpy是一个用于数值计算的Python库，具有高效的数组操作和广泛的数学函数。有时候，我们需要将Python中的数据结构保存到文件中以供稍后使用。在这种情况下，将对象数组写入.txt文件可能是一个好的选择。

阅读更多：Numpy 教程

步骤1：创建对象数组

首先，让我们创建一个包含学生姓名和成绩的对象数组。每个元素都是一个Python字典，其中包含’name’和’score’两个键和对应的值。

import numpy as np

students = np.array([
    {'name': 'Alice', 'score': 85},
    {'name': 'Bob', 'score': 90},
    {'name': 'Charlie', 'score': 95}
])

现在，我们有一个包含3个元素的对象数组。

步骤2：指定输出格式

要将对象数组写入.txt文件，我们需要将每个元素格式化为一个字符串，并将所有元素连接起来。我们可以使用Numpy的np.savetxt()函数实现这一目标。该函数需要指定输出文件名，数组，和输出格式。对于对象数组，我们需要指定输出格式为字符串，以便将字典格式化为文本字符串。

fmt = ['%s', '%d']

我们定义了一个包含两个占位符的格式字符串：第一个占位符使用’%s’表示字符串类型，第二个占位符使用’%d’表示整数类型。

步骤3：写入文件

现在，我们可以使用np.savetxt()函数将对象数组写入.txt文件。

np.savetxt('students.txt', students, fmt=fmt)

这行代码将对象数组students写入文件students.txt中，并使用fmt参数指定输出格式。

步骤4：读取文件

我们可以使用Python的文件读取函数open()和readlines()读取写入的文件。

with open('students.txt', 'r') as f:
    lines = f.readlines()

for line in lines:
    print(line.strip())

这段代码打印了写入的文件中的每一行字符串。

输出结果：

Alice   85
Bob     90
Charlie 95

总结

在本文中，我们学习了如何使用Numpy将对象数组写入.txt文件。我们首先创建了一个包含学生姓名和成绩的对象数组，然后定义了输出格式，接着将对象数组写入文件，并使用Python的文件读取函数读取了写入的文件。现在你可以自己尝试使用Numpy将其他Python数据结构写入文件。