Python CSV文件读取和写入

本文主要为Python 实现CSV文件读取和写入操作。

CSV文件写入和读取

因为没有现成的csv文件，所以csv的顺序为先写入后读取。

写入

创建csv文件并把数据写入，有两种实现方式：直接插入所有行和插入单行。

示例如下：

import csv
import osdef write_csv(path, data):with open(path, 'w') as f:writer = csv.writer(f)# 插入所有行# writer.writerows(data)# 循环插入单行for row in data:writer.writerow(row)path = os.path.join(os.getcwd(), 'test_1.csv')
data = [['id', '名称', '性别', '年龄'],['1', '张三', '男', '18'],['2', '李四', '女', '19'],['3', '王五', '女', '18'],
]
write_csv(path, data)

只是插入的数据会有换行情况，暂时没解决。

效果如下：

读取

读取上面写入的csv文件。

示例如下：

import csv
import osdef read_csv(path):infoList = []with open(path, 'r') as f:allInfo = csv.reader(f)for row in allInfo:infoList.append(row)return infoListpath = os.path.join(os.getcwd(), 'test_1.csv')
print(read_csv(path))

效果如下：

[['id', '名称', '性别', '年龄'], [], ['1', '张三', '男', '18'], [], ['2', '李四', '女', '19'], [], ['3', '王五', '女', '18'], []]

总结

本文主要为Python 实现CSV文件读取和写入操作，

对数据操作可在读取文件后进行相应判断处理。