python 往txt中不断写入数据

原创

mob64ca12db3721 2024-08-26 07:15:12 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12db3721的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python 通过文件写入不断记录数据的实践

在数据科学与编程领域，处理数据是一项基本而重要的技能。Python 作为一种强大的编程语言，提供了多种方法来处理数据，其中包括将数据写入文件。本文将探讨如何在 Python 中不断向 TXT 文件写入数据，并附上相关的代码示例，以帮助读者掌握这项技能。

文件操作基础

在 Python 中，文件操作主要包括打开文件、读写数据以及关闭文件。使用内置的 open() 方法可以方便地打开文件。文件模式包括：

'r'：只读模式
'w'：写入模式（会覆盖文件）
'a'：追加模式（在文件末尾添加数据）

在本示例中，我们将使用 'a' 模式，确保每次写入都会在原有内容后追加数据。

代码示例

下面是一个简单示例，演示了如何向 TXT 文件中不断写入数据。这个程序每隔一段时间就会写入当前的时间戳。

import time

def write_to_file(filename):
    with open(filename, 'a') as file:  # 以追加模式打开文件
        while True:
            current_time = time.ctime()  # 获取当前时间
            file.write(current_time + '\n')  # 写入当前时间
            file.flush()  # 刷新文件，确保数据被写入
            time.sleep(5)  # 等待5秒

if __name__ == "__main__":
    write_to_file('output.txt')

代码解析

在上面的代码中：

导入模块：首先，我们导入了 time 模块，以便我们可以获取当前的系统时间。
定义函数：定义 write_to_file 函数，接收一个文件名作为参数。
打开文件：使用 with open(filename, 'a') as file 语句，确保文件被正确打开和关闭。这里使用的是追加模式 'a'。
无限循环：使用 while True 创建一个无限循环，不断写入数据。
获取时间并写入：通过 time.ctime() 获取当前时间，使用 file.write() 将时间写入文件并换行。
刷新数据：使用 file.flush() 立即将缓冲区的数据写入文件，确保不会因为系统故障而丢失数据。
暂停：通过 time.sleep(5) 每5秒写入一次数据。

类图

在进行更复杂的文件写入操作时，我们可以将代码结构化，并使用类来封装功能。以下是一个简单的类图，展示了文件写入操作的一种设计思路。

classDiagram
    class FileWriter {
        +function __init__(filename)
        +function write_data(data)
        +function close_file()
    }

类的实现

基于上面的类图，我们可以实现一个 FileWriter 类，封装文件写入的逻辑。

class FileWriter:
    def __init__(self, filename):
        self.filename = filename
        self.file = open(self.filename, 'a')  # 打开文件

    def write_data(self, data):
        self.file.write(data + '\n')  # 写入数据
        self.file.flush()  # 刷新文件

    def close_file(self):
        self.file.close()  # 关闭文件

if __name__ == "__main__":
    writer = FileWriter('output.txt')
    try:
        while True:
            current_time = time.ctime()
            writer.write_data(current_time)
            time.sleep(5)
    except KeyboardInterrupt:
        print("停止写入")
    finally:
        writer.close_file()  # 确保文件被关闭