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

在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文件。