Python CSV文件读取和写入

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

CSV文件写入和读取

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

写入

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

示例如下：

import csv
import os


def 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 os

def read_csv(path):
    infoList = []
    with open(path, 'r') as f:
        allInfo = csv.reader(f)
        for row in allInfo:
            infoList.append(row)
        return infoList


path = os.path.join(os.getcwd(), 'test_1.csv')
print(read_csv(path))

效果如下：

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

总结

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

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