如何在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()方法来写入多行数据。
