python csv 文件的读取和写入

20 min read

读取 CSV 文件可以使用 Python 的内置 csv 模块,并调用其中的 reader() 函数。下面的示例代码演示了如何读取 CSV 文件:

import csv

# 打开 CSV 文件
with open('data.csv', newline='') as f:
    # 创建一个 csv 读取对象
    reader = csv.reader(f)

    # 遍历每一行数据
    for row in reader:
        # 输出每一行数据
        print(row)

写入 CSV 文件可以使用 csv 模块中的 writer() 函数。下面的示例代码演示了如何将数据写入 CSV 文件:

import csv

# 要写入 CSV 文件的数据
data = [
    ['Name', 'Country', 'Age'],
    ['John', 'USA', 25],
    ['Emily', 'UK', 30],
    ['Sam', 'Canada', 21]
]

# 打开或新建一个 CSV 文件
with open('output.csv', 'w', newline='') as f:
    # 创建一个 csv 写入对象
    writer = csv.writer(f)

    # 写入数据
    writer.writerows(data)

print('数据写入完成!')

以上代码会将数据写入名为 "output.csv" 的 CSV 文件中。