欢迎访问宙启技术站
智能推送

使用Python的CSV函数库读取和写入CSV文件

发布时间:2023-06-29 22:22:36

在Python中,可以使用csv库来读取和写入CSV文件。CSV(逗号分隔值)是一种常用的文件格式,它以逗号作为分隔符来存储表格数据。CSV文件通常用于数据交换和数据存储,因为它可以被各种程序和工具轻松地读取和处理。

首先,我们需要导入csv库:

import csv

读取CSV文件

一旦导入了csv库,我们就可以使用其提供的函数来读取CSV文件。最常用的函数是reader(),它可以读取CSV文件中的数据并返回一个列表。以下是读取CSV文件的基本方法:

# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建CSV读取器
    csv_reader = csv.reader(file)
    
    # 逐行读取CSV数据
    for row in csv_reader:
        # 使用row变量来访问每一行的数据
        print(row)

以上代码中,我们首先使用open()函数打开一个名为data.csv的CSV文件,并指定'r'作为文件的模式,表示读取文件。然后,我们使用csv.reader()函数创建一个CSV读取器,它将文件对象作为参数。最后,我们使用for循环逐行读取CSV数据,每一行数据都被存储在名为row的变量中,我们可以对其进行处理。

写入CSV文件

与读取CSV文件类似,我们可以使用csv库中的writer()函数来写入CSV文件。以下是写入CSV文件的基本方法:

# 创建CSV数据
data = [
    ['Name', 'Age', 'Gender'],
    ['John Smith', '25', 'Male'],
    ['Jane Doe', '30', 'Female'],
    ['Mike Johnson', '35', 'Male']
]

# 打开CSV文件
with open('data.csv', 'w', newline='') as file:
    # 创建CSV写入器
    csv_writer = csv.writer(file)
    
    # 写入CSV数据
    for row in data:
        csv_writer.writerow(row)

在以上代码中,我们首先创建了一个名为data的列表,其中包含了要写入CSV文件的数据。然后,我们使用open()函数打开一个名为data.csv的CSV文件,并指定'w'作为文件的模式,表示写入文件。接下来,我们使用csv.writer()函数创建一个CSV写入器,它将文件对象作为参数。最后,我们使用for循环逐行写入CSV数据,每一行数据被存储在名为row的变量中,我们通过调用csv_writer对象的writerow()函数将其写入CSV文件中。

总结

使用Python的csv库可以轻松地读取和写入CSV文件。通过使用csv.reader()函数可以逐行读取CSV文件中的数据,并通过使用csv.writer()函数可以逐行写入CSV文件中的数据。