如何使用Python将文本文件读入列表或数组？

在Python中，读取文本文件并将其存储在列表或数组中是一个很常见的任务。这种任务可以通过使用Python的内置函数和库来完成。

阅读更多：Python 教程

读取文本文件

在Python中，可以使用内置函数open()打开文本文件。open()函数有两个参数，第一个参数是文件名，第二个参数是打开文件的模式。打开文件的模式有如下几种：

• 'r' ： 只读模式
• 'w' ： 写入模式
• 'a' ： 追加模式
• 'x' ： 创建模式

默认情况下，open()函数是以只读模式打开文件的。当文件被打开后，可以使用内置函数read()读取文件中的内容。例如，以下代码将打开一个名为sample.txt的文本文件并将其内容读取到一个字符串中：

with open('sample.txt', 'r') as f:
    data = f.read()

在上面的代码中，with语句和as关键字组成了文件上下文管理器。使用文件上下文管理器是一种好的编程习惯，它可以确保文件在使用完成后会被正确关闭。将读取到的文本存储在字符串变量data中。

将文本转化为列表

将文本内容转化为列表是一个实用的技巧，因为它允许我们基于文本内容执行其他操作。例如，假设我们有一个包含名字的文本文件，我们可以将其读入列表中，并使用for循环遍历该列表以打印出每个名字。接下来，我们将解释如何将文本文件读入Python中的列表。

按行读取文本文件

通常，我们会将文本文件中的每一行读取到列表中。这个过程常常称为按行读取。以下代码演示了将文本文件中的行读取到Python列表中的基本方法：

with open('sample.txt') as f:
    lines = f.readlines()

在这个代码中，我们打开名为sample.txt的文本文件，并使用readlines()函数按行读取它的内容，每行都被存储在列表中。

读取并清理文本文件

通常，我们需要对读取的文本数据进行某些处理，例如去除换行符（\n）和空格符。以下代码演示了将读取的文本进行清理的方法：

with open('sample.txt') as f:
    lines = [line.strip() for line in f]

在这个代码中，我们使用简单的列表推导式将读取到的每一行进行了清理。上面的代码会排除行间的空格并删除的每一行的结尾处的换行符（\n）。

将读取的数据用逗号分隔

通常，在文本文件中，每一行都包含了一个或多个单词或数字，它们之间用逗号隔开。在这种情况下，我们可以使用.split()方法将每行文本数据按逗号拆分成单独的项。以下代码演示了如何将读取到的每一行按逗号拆分成列表：

with open('sample.txt') as f:
    lines = [line.strip().split(',') for line in f]

将文本转化为数组

在Python中，数组是一种用于存储和操作数值数据的数据结构。Python中的数组可以使用NumPy（Numerical Python）库来实现，NumPy是一个Python的扩展库，它拥有丰富的数学函数和内置数据结构。

安装NumPy库

在安装NumPy库之前，请确保您的Python版本是最新的。通过在终端或命令提示符下运行以下命令，可以安装NumPy库：

pip install numpy

将列表转换为NumPy数组

在安装了NumPy库之后，我们可以将Python列表转换为NumPy数组。以下代码演示了如何将Python列表转换为NumPy数组：

import numpy as np

my_list = [1, 2, 3, 4, 5]
my_array = np.array(my_list)

在这个代码中，我们使用NumPy库中的array()函数将Python列表转换为NumPy数组。这个函数创建一个新的NumPy数组，其中数据来自Python列表。我们还使用import语句导入了NumPy库并将其别名设置为np。

从文本文件创建NumPy数组

要将文本文件读入NumPy数组，首先需要将文本文件读取到Python列表中，然后将Python列表转换为NumPy数组。以下代码演示这个过程：

import numpy as np

with open('data.txt') as f:
    lines = [line.strip().split(',') for line in f]

data = np.array(lines, dtype=float)

在这个代码中，我们首先将名为data.txt的文本文件读取到Python列表lines中。然后，我们使用NumPy库中的array()函数将Python列表转换为NumPy数组。我们还使用dtype参数将数组中的元素类型设置为浮点数，在这个示例中，我们将其设置为float。

结论

Python提供了读取文本文件并将其存储在列表或数组中的便捷方法。对于简单的文本文件读取任务，可以使用简单的Python内置函数和语法来实现。对于更复杂的数值数据处理任务，可以使用NumPy库来创建和操作NumPy数组。同时，对于任何文本处理任务，请确保文件在使用后被正确地关闭。