本文主要为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文件读取和写入操作,
对数据操作可在读取文件后进行相应判断处理。