拥有自己的Python-CSV模块

在工作中，如果你经常需要读写CSV文本文件，恰好你又会python，那你肯定会使用python中的CSV模块，使用CSV模块读写CSV文件确实轻松了很多。使用最多的两个方法就是 csv.reader() 和 csv.writer()。由于每次读写都需要编写相同的琐碎代码，对于列的处理，还需要代码拆解实现。所以可以将这些琐碎的方法封装起来，变为更贴合自己的工具。

CSV模块的自定义封装

封装此类需要具有的功能：

1. 读取CSV文件

按行读取全列数据
按行读取指定列名的数据
获取CSV数据的行数(从指定的标题列开始数)

2. 写CSV文件

可以指定列表头写入

类代码

下面只列出了功能方法名，代码实现由于空间问题就省略了。具体代码可以下载附件查看。

class CsvOpt:
    def __init__(self, file_name:str, rwmode='r', headline:int=1):
    def read_file(self):---
    def getitem(self, row:int, item:str): ---
    def getrow(self, row:int):---
    def get_maxrow(self):---
    def write_file(self, data_lines:list, head_list:list=[]):---

方法介绍

__ init__(self, file_name:str, rwmode=‘r’, headline:int=1)
file_name：要操作的csv文件名。
rwmode：操作文件的读写模式，即 open() 方法中使用的模式，默认为读模式，即 “r”。
headline：指定csv文件中列头的行号，默认使用第一行作为表头。
作用：初始化
read_file(self)
无参函数，作用为读取整个文件内容
getitem(self, row:int, item:str)
row: 要读取的行号，
item: 要读取的列的名字
作用：读取第row行，列名为item的单元的值
getrow(self, row:int)
row：行号
作用：获取第row行的所有数据
get_maxrow(self)
无参函数，返回csv文件从列头位置开始数的行数。
write_file(self, data_lines:list, head_list:list=[])
data_lines：数据列表，例如：[[行数据],[行数据],[行数据],[行数据]]
head_list：指定的列头：例如：[‘姓名’,‘年龄’,‘性别’,‘位置’]，默认为空
作用：写数据到指定的csv文件中

使用案例介绍

读取文件

文件名：test.csv

姓名	年龄	地址
张三	17	深圳
李四	18	佛山
周五	19	山西
郑王	20	上海

# -*- coding:utf-8 -*-

from csvopt import CsvOpt

f = "D:\\csvtest\\test.csv"
mycsv = CsvOpt(f) # 初始化文件
mycsv.read_file() # 读取文件

print('按行和指定列名输出:')
for i in range(1, mycsv.get_maxrow() + 1):
    print(mycsv.getitem(i, '姓名'), mycsv.getitem(i, '年龄'))

print('按行输出所有列:')
for i in range(1, mycsv.get_maxrow() + 1):
    print(mycsv.getrow(i))

输出如下：

按行和指定列名输出:
张三 17
李四 18
周五 19
郑王 20
按行输出所有列:
['张三', '17', '深圳']
['李四', '18', '佛山']
['周五', '19', '山西']
['郑王', '20', '上海']

读取指定表头的文件