使用Python创建CSV文件并写入数据

VTRNJ • 2025-04-28 13:17 • 编程

本文将介绍如何使用Python创建CSV文件并写入数据，主要包括以下几个方面：

一、CSV文件介绍

CSV(Comma-Separated Values)文件是一种常用的简单数据交换格式，通常用于把表格数据从一种应用程序转移到另一种应用程序。CSV文件由文本行组成，每行表示一个数据记录，每个字段用逗号分隔。CSV文件可以使用任何文本编辑器打开，很容易存储和处理。

二、创建CSV文件

Python内置的csv模块提供了一些方法来创建CSV文件，其中最常用的是csv.writer()。csv.writer()将数据写入文件对象，并自动加上逗号和换行符。以下是创建CSV文件的简单示例：

import csv

with open('example.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile, delimiter=',')
    writer.writerow(['Name', 'Age', 'City'])
    writer.writerow(['Tom', '28', 'Beijing'])
    writer.writerow(['Jack', '30', 'Shanghai'])

以上代码创建了一个名为example.csv的CSV文件，写入了三个数据记录，每个记录包括Name、Age和City三个字段。写入文件的过程使用了with语句，该语句可以自动关闭文件，确保写入数据的完整性。

三、读取CSV文件

除了创建CSV文件外，Python的csv模块还提供了一些方法来读取CSV文件。csv.reader()方法可以将CSV文件转换为Python的列表或者字典形式，方便后续的数据处理和分析。以下是读取CSV文件的示例：

import csv

with open('example.csv', newline='') as csvfile:
    reader = csv.reader(csvfile, delimiter=',')
    for row in reader:
        print(', '.join(row))

以上代码使用csv.reader()方法读取了example.csv文件并逐行打印。csv.reader()的delimiter参数指定了字段分隔符，默认为’,’。

四、写入字典数据

除了写入列表数据，我们还可以将字典数据写入CSV文件。以下是写入字典数据的示例：

import csv

with open('example_dict.csv', 'w', newline='') as csvfile:
    fieldnames = ['Name', 'Age', 'City']
    writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerow({'Name': 'Tom', 'Age': '28', 'City': 'Beijing'})
    writer.writerow({'Name': 'Jack', 'Age': '30', 'City': 'Shanghai'})

以上代码创建了一个名为example_dict.csv的CSV文件，写入了两个数据记录，每个记录包括Name、Age和City三个字段。写入文件的过程使用了DictWriter类，它允许我们通过字典的形式来写入CSV内容。需要特别注意的是，在写入数据前需要使用writeheader()方法写入表头信息。

五、读取字典数据

和写入字典数据类似，我们也可以使用csv.DictReader()方法来读取CSV文件中的字典数据。以下是读取字典数据的示例：

import csv

with open('example_dict.csv', newline='') as csvfile:
    reader = csv.DictReader(csvfile)
    for row in reader:
        print(row['Name'], row['Age'], row['City'])

以上代码使用csv.DictReader()方法读取了example_dict.csv文件并逐行打印。csv.DictReader()方法将每一行数据转换为一个字典，其中字典的键对应于表头信息，字典的值对应于每个字段的值。

六、总结

本文介绍了如何使用Python创建CSV文件并写入数据，以及读取CSV文件的方法。通过学习本文，你可以使用Python轻松地进行CSV文件数据的读写和处理，实现数据的快速转换和分析。

原创文章，作者：VTRNJ，如若转载，请注明出处：https://www.506064.com/n/375011.html