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

使用Python的open()函数读取和写入CSV文件

发布时间:2023-12-16 03:13:46

Python的open()函数是一种在文件操作中常用的函数,它可以用于读取和写入CSV文件。

读取CSV文件:

要读取CSV文件,可以使用open()函数打开文件并指定读取模式。然后,可以使用csv模块的reader()函数来创建一个读取器对象,该对象可以逐行读取CSV文件的内容。

下面是一个读取CSV文件的例子:

import csv

# 打开CSV文件
with open('data.csv', 'r') as file:
    # 创建一个读取器对象
    reader = csv.reader(file)

    # 逐行读取CSV文件
    for row in reader:
        # 处理每一行的数据
        print(row)

上面的例子中,使用open()函数以读取模式打开名为'data.csv'的文件。然后使用csv模块的reader()函数创建了一个读取器对象,并将其赋值给变量reader。接下来,使用一个for循环来逐行读取CSV文件的内容,并将每一行打印出来。

写入CSV文件:

首先,要写入CSV文件,需要以写入模式使用open()函数打开一个文件。然后可以使用csv模块的writer()函数来创建一个写入器对象,该对象可以将数据写入CSV文件中。

下面是一个写入CSV文件的例子:

import csv

# 创建要写入的数据
data = [['Name', 'Age', 'Country'],
        ['John', '25', 'USA'],
        ['Jane', '30', 'Canada'],
        ['Tom', '35', 'UK']]

# 打开CSV文件
with open('output.csv', 'w', newline='') as file:
    # 创建一个写入器对象
    writer = csv.writer(file)

    # 写入数据到CSV文件
    writer.writerows(data)

print('数据写入完成')

上面的例子中,我们创建了一个名为data的列表,其中包含了要写入CSV文件的数据。然后使用open()函数以写入模式打开名为'output.csv'的文件,并指定newline参数为'',这是因为CSV文件中的每一行都会被默认地用空行分隔。接下来,使用csv模块的writer()函数创建了一个写入器对象,并将其赋值给变量writer。最后,通过调用writerows()函数写入数据到CSV文件中,该函数将整个列表的数据一次性写入。

总结:

open()函数是Python中用于文件操作的内置函数,可以被用来读取和写入CSV文件。在读取CSV文件时,我们可以使用csv模块的reader()函数创建一个读取器对象,并使用for循环逐行处理CSV文件的数据。在写入CSV文件时,我们可以使用csv模块的writer()函数创建一个写入器对象,并调用writerows()函数将数据一次性写入CSV文件中。