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

如何在Python中读取和写入CSV文件

发布时间:2023-12-04 08:38:47

在Python中,您可以使用csv模块来读取和写入CSV文件。CSV文件是一种常用的数据格式,以逗号分隔字段。

读取CSV文件:

为了读取CSV文件,您需要使用csv.reader()函数,并提供文件对象作为参数。以下是一个示例代码来读取名为"example.csv"的CSV文件,并打印每一行的字段:

import csv

with open('example.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

上述代码使用了with语句和csv.reader()函数来读取CSV文件。csv.reader()会将文件对象解析为一个CSV的迭代器,然后您可以用for循环遍历每一行,row变量则包含了该行的字段列表。

写入CSV文件:

为了写入CSV文件,您需要使用csv.writer()函数,并提供文件对象和一些数据作为参数。以下是一个示例代码来将一些数据写入名为"example.csv"的CSV文件:

import csv

data = [
    ['Name', 'Age', 'Country'],
    ['John', '25', 'USA'],
    ['Emily', '30', 'Canada'],
    ['Tom', '35', 'UK']
]

with open('example.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

上述代码使用了with语句和csv.writer()函数来写入CSV文件。csv.writer()会将文件对象解析为一个CSV的写入器,然后您可以使用writerow()方法写入一行数据,或者使用writerows()方法写入多行数据。

在上述例子中,newline=''参数是必需的,以确保在不同操作系统上的兼容性。

总结:

CSV文件是一种常用的数据格式,在Python中可以使用csv模块来读取和写入CSV文件。读取CSV文件时,使用csv.reader()函数和一个文件对象,然后可以使用for循环来遍历每一行数据。写入CSV文件时,使用csv.writer()函数和一个文件对象,然后可以使用writerow()方法来写入一行数据,或者使用writerows()方法来写入多行数据。