Python CSV 文件读写代码示例

当然！以下是一个用 Python 读取和写入 CSV 文件的示例代码：

import csv

# 读取 CSV 文件
def read_csv(filename):
    data = []
    with open(filename, 'r') as file:
        csv_reader = csv.reader(file)
        for row in csv_reader:
            data.append(row)
    return data

# 写入 CSV 文件
def write_csv(filename, data):
    with open(filename, 'w', newline='') as file:
        csv_writer = csv.writer(file)
        for row in data:
            csv_writer.writerow(row)
    print('CSV 文件写入完成！')

# 测试读取 CSV 文件
csv_data = read_csv('data.csv')
print(csv_data)

# 测试写入 CSV 文件
new_data = [['Name', 'Age', 'Country'],
            ['John', '25', 'USA'],
            ['Alice', '30', 'UK'],
            ['Bob', '35', 'Canada']]
write_csv('new_data.csv', new_data)

在这个例子中，read_csv 函数使用 csv.reader 来读取指定文件名的 CSV 文件，并将每一行数据添加到一个列表中，最后返回完整的数据列表。

write_csv 函数接受一个文件名和一个数据列表作为参数，并使用 csv.writer 来将数据逐行写入到一个新的 CSV 文件中。

请注意，在写入 CSV 文件时，需要指定 newline='' 参数，以确保在 Windows 系统上正确处理换行符。

最后，我们通过调用这两个函数来演示如何读取和写入 CSV 文件。

希望这个例子对你有所帮助！如有任何其他问题，请随时提问。