将 CSV 读取为列表 Python

在 Python 中将 CSV（逗号分隔值）文件读取为列表可以使用内置的 csv 模块。csv 模块提供了快速解析和写入 CSV 文件的功能。

读取 CSV 文件

使用 csv 模块的 reader() 函数可以将 CSV 文件读取为列表。

import csv

with open('file.csv') as csv_file:
    csv_reader = csv.reader(csv_file, delimiter=',')
    data = []
    for row in csv_reader:
        data.append(row)

解释：

• 首先导入 csv 模块
• 然后用 open() 函数打开 CSV 文件，并将其分配给 csv_file 变量
• 确定分隔符。由于 CSV 文件中逗号是默认分隔符，因此 csv.reader() 函数不需要显式指定。如果需要使用其他分隔符，可以在 delimiter 参数中指定
• 创建一个空列表 data 以存储 CSV 文件的数据
• 使用 for 循环迭代 csv_reader 变量来获取 CSV 文件中的所有行
• 将每一行添加到 data 列表中

现在，data 列表包含了 CSV 文件中的行数据。

写入 CSV 文件

使用 csv 模块的 writer() 函数可以将列表写入 CSV 文件。

import csv

data = [['Name', 'Age', 'Gender'], ['Tom', '25', 'Male'], ['Jane', '30', 'Female']]

with open('file.csv', mode='w', newline='') as csv_file:
    writer = csv.writer(csv_file)
    for row in data:
        writer.writerow(row)

解释：

• 首先导入 csv 模块
• 创建一个包含数据的列表 data
• 使用 open() 函数以写入模式打开 CSV 文件，并将其分配给 csv_file 变量。newline='' 参数用于避免出现额外的空行
• 创建一个 csv 写入器 writer 并将其分配给 writer 变量
• 使用 for 循环迭代 data 列表中的所有行
• 将每行写入 CSV 文件中

现在，CSV 文件包含了 data 列表中的所有行数据。

总结

CSV 是一种常见的数据格式，在 Python 中读取和写入 CSV 文件非常容易。可以使用 csv 模块的 reader() 和 writer() 函数来实现。