【Python数据分析系列】实现txt文件与列表(list)相互读写转换

off999 2024-10-04 00:19 55 浏览 0 评论

这是Python数据分析系列原创文章，我的第199篇原创文章。

一、问题

平时在做数据分析或者程序开发的时候，需要将中间的一些结果或最后的处理结果保存下来，比如保存为txt格式的文本文件，这就涉及列表与txt之间的一种读取和写入操作，本文将用一个简单的例子来实现。

二、一维列表与txt读写转换

2.1 一维列表写入txt文件

将一维列表data中的每个元素写入文件中，每个元素占一行：

data = [1, 2, 3, 4, 5]


with open("output1.txt", "w") as file:
    for item in data:
        file.write(str(item) + "\n")

上述代码将创建一个名为output1.txt的文本文件，并将列表data中的每个元素写入文件中，每个元素占一行。open("output1.txt", "w")用于打开文件，"w"表示以写入模式打开文件。with open(...) as file:语句用于自动管理文件的打开和关闭，确保在使用完文件后正确关闭它。

for item in data循环遍历列表中的每个元素，并使用file.write(...)将每个元素写入文件。str(item)将元素转换为字符串形式，"\n"表示换行符，用于将每个元素写入文件的不同行。

执行上述代码后，将在当前目录下创建一个名为output.txt的文本文件，并将列表中的元素写入文件中。

需要注意的是，如果列表中的元素不是字符串类型，需要使用适当的方法将其转换为字符串形式，以便写入文件。在上面的示例中，使用str(item)将元素转换为字符串。

2.2 读取txt文件为一维列表

逐行读取文件内容，写入列表：

with open("output1.txt", "r") as file:
    data = file.read().splitlines()


print(data)

首先使用open("output1.txt", "r")打开名为input.txt的文本文件，以读取模式打开文件。然后，file.read()方法将整个文件内容作为一个字符串读取出来。接下来，splitlines()方法将字符串按行分割，并返回一个包含每行内容的列表。

执行上述代码后，将读取名为output1.txt的文本文件，并将文件中的每一行作为列表的一个元素存储在data中。

请注意，如果文件非常大，一次性读取整个文件内容可能会占用大量内存。在处理大型文件时，可能需要考虑逐行读取或使用其他适当的方法来处理文件内容。

三、二维列表与txt读写转换

3.1 二维列表写入txt文件

将二维列表data中的每个元素写入文件中，每个元素占一行：

data = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]


with open("output2.txt", "w") as file:


    for row in data:
        for item in row:
            file.write(str(item) + ' ')
        file.write('\n')

运行以上代码后，将会在当前目录下生成一个名为"output2.txt"的文件，其中包含二维列表的数据。每行表示二维列表中的一行，每个元素之间使用空格分隔。请注意，上述代码将数据以空格分隔并写入文件，如果你需要使用其他分隔符，可以在write()方法中修改。

3.2 读取txt文件为二维列表

读取txt文件，输出为二维列表，每个元素为一行：

with open('output2.txt', 'r') as file:
    lines = file.readlines()
    data = []
    for line in lines:
        row = line.strip().split(' ')
        data.append(row)

lines：

data：

使用readlines()方法读取文件的所有行，并将其存储在一个列表中。遍历行列表，使用split()方法将每行按照指定的分隔符（如空格）分割为多个元素，并将其存储在二维列表中。请注意，上述代码假设每行数据以空格分隔。如果你的文件中使用其他分隔符，可以在split()方法中修改分隔符参数。另外，strip()方法用于去除每行开头和结尾的空白字符，以确保数据的准确性。