python 如何写csv文件内容

未希 • 2024-04-09 08:29 • 技术教程 • 阅读 99

在Python中，我们可以使用csv模块来写入CSV文件，CSV（CommaSeparated Values）是一种常见的数据存储格式，它以逗号分隔每个字段的值，以下是如何使用Python编写CSV文件的详细教程。

（图片来源网络，侵删）

1、导入csv模块

我们需要导入csv模块，这个模块包含了一些用于处理CSV文件的函数和方法。

import csv

2、准备数据

在写入CSV文件之前，我们需要准备一些数据，这些数据可以是列表、元组或其他任何可迭代对象，我们可以创建一个包含姓名和年龄的字典列表：

data = [
    {"name": "张三", "age": 25},
    {"name": "李四", "age": 30},
    {"name": "王五", "age": 35},
]

3、打开文件并创建写入器

接下来，我们需要打开一个文件并将它与一个写入器对象关联起来，我们可以使用open()函数以写入模式（’w’）打开文件，然后使用csv模块的writer()函数创建一个写入器对象。

with open("output.csv", "w", newline="", encoding="utf8") as csvfile:
    writer = csv.DictWriter(csvfile, fieldnames=["name", "age"])

这里，我们使用with语句来确保文件在操作完成后被正确关闭，我们还指定了newline=""参数来避免在Windows系统中出现额外的空行，以及encoding="utf8"参数来确保文件中的中文字符能够正确显示。

4、写入数据

现在，我们可以使用写入器的writeheader()方法来写入字段名，然后使用writerow()或writerows()方法来写入数据，我们可以先写入字段名，然后逐行写入数据：

写入字段名
writer.writeheader()
逐行写入数据
for row in data:
    writer.writerow(row)

或者，我们可以先将所有数据转换为一个列表，然后一次性写入：

将所有数据转换为一个列表
data_list = [item for sublist in data for item in sublist.values()]
一次性写入数据
writer.writerows(zip(*data))

这里，我们使用zip()函数将每个字典的值组合成一个元组，然后使用*操作符将这些元组展开为一个列表，我们将这个列表传递给writerows()方法进行写入。

5、完整示例

将以上步骤组合在一起，我们可以得到一个完整的示例：

import csv
data = [
    {"name": "张三", "age": 25},
    {"name": "李四", "age": 30},
    {"name": "王五", "age": 35},
]
with open("output.csv", "w", newline="", encoding="utf8") as csvfile:
    writer = csv.DictWriter(csvfile, fieldnames=["name", "age"])
    writer.writeheader()
    for row in data:
        writer.writerow(row)

运行这段代码后，当前目录下会生成一个名为output.csv的文件，其中包含了我们刚刚写入的数据，我们可以使用Excel或其他表格处理软件打开这个文件，查看其内容。

原创文章，作者：未希，如若转载，请注明出处：https://www.kdun.com/ask/455397.html