python 操作csv文件

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

# -*- coding: utf-8 -*-
import csv


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

print read_csv("test.csv")