在Python中如何读取和写入CSV文件
发布时间:2023-12-25 19:05:39
在Python中,可以使用csv模块来读取和写入CSV文件。
1. 读取CSV文件:
要读取CSV文件,可以使用csv.reader类。以下是一个读取CSV文件的示例代码:
import csv
# 打开CSV文件
with open('data.csv', 'r') as file:
# 创建一个csv.reader对象
csv_reader = csv.reader(file)
# 逐行读取数据
for row in csv_reader:
# 打印每行的数据
print(row)
在上述示例中,打开文件并创建csv.reader对象后,可以使用for循环逐行读取数据。每行数据通过列表的形式返回,其中每个元素代表CSV文件中的一个单元格。
2. 写入CSV文件:
要写入CSV文件,可以使用csv.writer类。以下是一个写入CSV文件的示例代码:
import csv
# 要写入的数据
data = [
['Name', 'Age', 'City'],
['John', 28, 'New York'],
['Lisa', 32, 'London'],
['Tom', 45, 'Tokyo']
]
# 打开CSV文件
with open('output.csv', 'w', newline='') as file:
# 创建一个csv.writer对象
csv_writer = csv.writer(file)
# 写入数据
csv_writer.writerows(data)
在上述示例中,打开文件并创建csv.writer对象后,可以使用writerows方法写入数据。writerows方法接受一个列表,每个元素代表CSV文件中的一行数据。在这个例子中,我们写入了一个包含多个行的二维列表。
注意,在写入CSV文件时,newline=''参数是必须的,否则可能会出现空行问题。
总结:
使用csv模块可以方便地读取和写入CSV文件,它提供了一组方法来处理CSV数据。读取CSV文件时,可以使用csv.reader类逐行读取数据;写入CSV文件时,可以使用csv.writer类将数据写入文件。以上是简单的示例,你可以根据实际需求来读取和写入更复杂的CSV文件。
