用Python编写 CSV 文件

CSV 文件是一种常见的文本文件格式，它用逗号分隔不同数据字段，每一行代表一个记录。Python 提供了 csv 模块来读写 CSV 文件。

写入 CSV 文件

要写入 CSV 文件，可以使用 csv.writer 对象。首先需要导入 csv 模块，然后创建一个 csv.writer 对象并使用其 writerow 方法写入行数据。

import csv

# 创建一个 CSV 文件写入器对象
csvfile = open('data.csv', 'w', newline='')
csvwriter = csv.writer(csvfile)

# 写入表头
csvwriter.writerow(['姓名', '年龄', '性别'])

# 写入数据
csvwriter.writerow(['张三', '18', '男'])
csvwriter.writerow(['李四', '20', '女'])

# 关闭文件
csvfile.close()

上面的代码创建了一个名为 data.csv 的 CSV 文件，并写入了表头和两行数据。

在创建 csv.writer 对象时，需要指定文件名、写入模式和 newline 参数。newline 参数用于控制写入行的结束符，它的默认值是 \n，指定空字符串表示不使用结束符。

读取 CSV 文件

要读取 CSV 文件，也可以使用 csv 模块。使用 csv.reader 对象可以轻松地逐行读取 CSV 文件的内容。

import csv

# 打开 CSV 文件并读取内容
with open('data.csv', newline='') as csvfile:
    csvreader = csv.reader(csvfile)

    # 跳过表头
    next(csvreader)

    # 读取数据
    for row in csvreader:
        print('{0}: {1} 岁，{2}'.format(row[0], row[1], row[2]))

上面的代码使用 with 语句打开 data.csv 文件，并使用 csv.reader 对象读取文件内容。由于第一行是表头，可以使用 next 方法跳过。然后使用 for 循环逐行读取数据，并将其打印到控制台上。

总结

CSV 文件是一种常见的文本文件格式，Python 提供了 csv 模块来读写 CSV 文件。通过 csv.writer 对象可以轻松地写入 CSV 文件，通过 csv.reader 对象可以轻松地读取 CSV 文件。