Python CSV数据的写入和读取

16 min read

Python内置的CSV模块提供了一个简单易用的方式来写入和读取CSV文件。下面是一个简单的示例:

写入CSV文件

import csv

# 写入CSV文件
with open('file.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerow(['Name', 'Age', 'City'])
    writer.writerow(['Alice', '27', 'New York'])
    writer.writerow(['Bob', '25', 'San Francisco'])

上面的代码将创建一个名为file.csv的CSV文件,并写入三行数据到该文件中。mode='w'表示以写入模式打开文件,newline=''表示在写入CSV文件时不自动插入额外的空行。

读取CSV文件

import csv

# 读取CSV文件
with open('file.csv', mode='r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

上面的代码将读取名为file.csv的CSV文件,并逐行输出文件中的内容。mode='r'表示以只读模式打开文件。每行数据将被转换成一个列表,其中每个元素对应于CSV文件中的一个单元格。在上面的示例中,输出如下:

['Name', 'Age', 'City']
['Alice', '27', 'New York']
['Bob', '25', 'San Francisco']

以上便是Python操作CSV数据的简单例子,您可以参考Python官方文档以了解更多操作细节。