Python写入txt文件的几种方法

Python写入txt文件的几种方法

在Python中，我们可以使用多种方法将数据写入txt文件。无论是将程序运行的结果保存到文件中，还是将一些数据存储起来供后续使用，都可以使用下面介绍的方法来实现。

方法一：使用文件对象的write方法

示例代码：

# 打开或创建一个txt文件
file = open("output.txt", "w")

# 写入数据
file.write("Hello, World!\n")
file.write("Python写入txt文件的几种方法\n")

# 关闭文件
file.close()

运行结果：在当前目录下生成一个名为output.txt的文件，内容为：

Hello, World!
Python写入txt文件的几种方法

方法二：使用with语句自动关闭文件

示例代码：

# 使用with语句打开文件，不需要手动关闭文件
with open("output.txt", "w") as file:
    file.write("Hello, World!\n")
    file.write("Python写入txt文件的几种方法\n")

运行结果：同样会在当前目录下生成一个名为output.txt的文件，内容与上述相同。

方法三：使用csv模块写入csv格式文件

示例代码：

import csv

data = [
    ['Name', 'Age', 'Country'],
    ['John', '28', 'USA'],
    ['Emma', '25', 'Canada'],
    ['Mike', '30', 'UK'],
]

# 打开或创建一个csv文件
with open('output.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerows(data)

运行结果：在当前目录下生成一个名为output.csv的文件，内容为：

Name,Age,Country
John,28,USA
Emma,25,Canada
Mike,30,UK

方法四：使用pandas库写入csv格式文件

示例代码：

import pandas as pd

data = {
    'Name': ['John', 'Emma', 'Mike'],
    'Age': [28, 25, 30],
    'Country': ['USA', 'Canada', 'UK']
}

df = pd.DataFrame(data)

# 将DataFrame写入csv文件
df.to_csv('output.csv', index=False)

运行结果：同样会在当前目录下生成一个名为output.csv的文件，内容与上述相同。

方法五：使用json模块写入json格式文件

示例代码：

import json

data = {
    'Name': 'John',
    'Age': 28,
    'Country': 'USA'
}

# 打开或创建一个json文件
with open('output.json', 'w') as jsonfile:
    json.dump(data, jsonfile)

运行结果：在当前目录下生成一个名为output.json的文件，内容为：

{"Name": "John", "Age": 28, "Country": "USA"}

这就是几种常见的Python写入txt文件的方法。无论是简单的文本文件，还是复杂的csv或json格式文件，Python都提供了相应的方法来满足我们的需求。根据具体情况选择合适的方法，可以使我们的代码更加简洁和高效。