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

Python如何读取和写入CSV文件的函数?

发布时间:2023-08-22 07:08:31

在Python中,可以使用csv模块来读取和写入CSV文件。

要读取CSV文件,我们可以使用csv.reader函数。首先,需要打开CSV文件,然后使用csv.reader函数来创建一个reader对象。可以使用reader对象的readrows()方法来逐行读取CSV文件中的数据。以下是一个例子:

import csv

# 打开CSV文件
with open('file.csv', 'r') as file:
    # 创建reader对象
    reader = csv.reader(file)
    
    # 读取数据
    for row in reader:
        print(row)

要写入CSV文件,我们可以使用csv.writer函数。首先,需要打开CSV文件,然后使用csv.writer函数来创建一个writer对象。可以使用writer对象的writerow()方法来逐行写入数据。以下是一个例子:

import csv

# 要写入的数据
data = [['Name', 'Age'], ['John Doe', 30], ['Jane Smith', 25]]

# 打开CSV文件
with open('file.csv', 'w', newline='') as file:
    # 创建writer对象
    writer = csv.writer(file)
    
    # 写入数据
    writer.writerows(data)

值得注意的是,在写入CSV文件时,需要使用newline=''参数来避免插入空行。

可以使用csv模块的其他函数和方法来处理CSV文件,如csv.DictReadercsv.DictWriter来处理包含标题行的CSV文件,和csv.writerwriterow()writerows()方法来逐行写入数据。

总结起来,要读取和写入CSV文件,可以使用csv.readercsv.writer函数,它们提供了方便的方法来处理CSV文件中的数据。