python 操作csv文件

这边主要指明对csv文件的读写操作， 欢迎留言

# -*- coding: utf-8 -*-import csvdef write_onerow(file_name, data, mode='wb', encoding='utf-8'):    '''    写入一行数据    :param file_name: 文件名    :param data:   一列表 或元祖    :param mode: 模式    :param encoding: 编码    :return:    '''    with open(file_name, mode, encoding=encoding) as f:        wri = csv.writer(f)        wri.writerow(data)def write_rows(file_name, data, mode="wb", encoding='utf-8'):    '''     写入多行数据  数组    :param file_name: 文件名    :param data: 数组 （列表， 元祖）    :param mode: 模式    :param encoding: 编码    :return:    '''    with open(file_name, mode, encoding=encoding) as f:        wri = csv.writer(f)        wri.writerows(data)def read_csv(file_name, mode="rb", encoding='utf-8'):    '''    :param file_name: 文件名    :param mode: 模式    :return:    返回数组  中包含列表    :param encoding: 编码    '''    with open(file_name, mode, encoding=encoding) as f:        ret = list()        reader = csv.reader(f)        for row in reader:            ret.append(row)        return retprint read_csv("test.csv")