Python实战教学之使用CSV数据

程序猿～厾罗2024-01-03 9:14

首先，我们需要导入csv模块：

python 复制代码

import csv

然后，我们可以通过以下方式打开CSV文件：

python 复制代码

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

这段代码会逐行读取CSV文件中的数据，并逐行打印出来。

接下来，我们可以根据实际需求对数据进行处理。比如，我们可以将数据存储到一个列表中：

复制代码

`data = []

with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        data.append(row)
`

这样，我们就将CSV文件中的数据存储到了一个名为data的列表中。

除了读取数据，我们还可以使用csv模块来写入数据到CSV文件中。比如，我们可以将一个列表中的数据写入CSV文件中：

复制代码

`data = [['Name', 'Age', 'Gender'], ['John', '20', 'Male'], ['Lisa', '22', 'Female']]

with open('output.csv', 'w') as file:
    writer = csv.writer(file)
    for row in data:
        writer.writerow(row)
`

这段代码会将data列表中的数据写入到output.csv文件中。