Python写入文件方法|极客教程

Python写入文件方法

在Python中，我们经常需要将程序处理过的数据保存到文件中，这样可以方便后续的使用或者备份。本文将详细介绍Python中写入文件的方法，包括文本文件、CSV文件、JSON文件等常见文件格式的写入操作。

写入文本文件

写入文本文件是最常见的文件操作之一，可以使用Python内置的open()函数来打开一个文件并写入数据。下面是一个简单的示例代码：

# 打开一个文本文件
file_path = "test.txt"
f = open(file_path, "w")

# 写入数据
f.write("Hello, world!\n")
f.write("This is a text file.\n")

# 关闭文件
f.close()

运行以上代码后，会在当前目录下生成一个名为test.txt的文本文件，文件内容为：

Hello, world!
This is a text file.

需要注意的是，使用open()函数时需要指定文件的打开模式，其中"w"表示以写入模式打开文件。如果文件不存在，open()函数会新建一个文件；如果文件已存在，原文件内容将被覆盖。

写入CSV文件

CSV（Comma-Separated Values）是一种常见的数据存储格式，可以使用Python的csv模块来进行CSV文件的读写操作。下面是一个简单的示例代码：

import csv

# 将数据写入CSV文件
data = [
    ['Alice', 25, 'Engineer'],
    ['Bob', 30, 'Manager'],
    ['Charlie', 35, 'CEO']
]

file_path = "data.csv"
with open(file_path, "w", newline='') as csvfile:
    csvwriter = csv.writer(csvfile)
    csvwriter.writerows(data)

运行以上代码后，会在当前目录下生成一个名为data.csv的CSV文件，文件内容如下：

Alice,25,Engineer
Bob,30,Manager
Charlie,35,CEO

在使用csv.writer()写入CSV文件时，需要传入newline=''参数，这是为了避免在Windows系统下出现空行。

写入JSON文件

JSON（JavaScript Object Notation）是一种轻量级的数据交换格式，Python内置的json模块可以用来操作JSON文件。下面是一个简单的示例代码：

import json

# 将数据写入JSON文件
data = {
    "name": "Alice",
    "age": 25,
    "occupation": "Engineer"
}

file_path = "data.json"
with open(file_path, "w") as jsonfile:
    json.dump(data, jsonfile)

运行以上代码后，会在当前目录下生成一个名为data.json的JSON文件，文件内容如下：

{"name": "Alice", "age": 25, "occupation": "Engineer"}

使用json.dump()方法将Python数据写入JSON文件时，会自动将数据转换为JSON格式进行保存。

写入其他文件格式

除了文本文件、CSV文件和JSON文件，Python还支持写入其他各种文件格式，比如Excel文件、SQLite数据库等。我们可以使用第三方库（如openpyxl、pandas、sqlite3等）来进行相应文件格式的写入操作。

以写入Excel文件为例，我们可以使用openpyxl库来进行处理。下面是一个简单的示例代码：

from openpyxl import Workbook

# 创建一个Excel工作簿并写入数据
wb = Workbook()
ws = wb.active
ws['A1'] = 'Name'
ws['B1'] = 'Age'
ws['C1'] = 'Occupation'

data = [
    ['Alice', 25, 'Engineer'],
    ['Bob', 30, 'Manager'],
    ['Charlie', 35, 'CEO']
]

for row in data:
    ws.append(row)

file_path = "data.xlsx"
wb.save(file_path)

运行以上代码后，会在当前目录下生成一个名为data.xlsx的Excel文件，文件内容如下：