CSV模块

Python csv模块的使用

1、csv简介

CSV (Comma Separated Values)，即逗号分隔值（也称字符分隔值，因为分隔符可以不是逗号），是一种常用的文本

格式，用以存储表格数据，包括数字或者字符。很多程序在处理数据时都会碰到csv这种格式的文件，它的使用是比

较广泛的（Kaggle上一些题目提供的数据就是csv格式），csv虽然使用广泛，但却没有通用的标准，所以在处理csv

格式时常常会碰到麻烦，幸好python内置了csv模块。下面简单介绍csv模块中最常用的一些函数。

更多内容请参考：

https://docs.python.org/2/library/csv.html#module-csv

https://docs.python.org/3/library/csv.html#module-csv

2、csv模块中的函数

• reader(csvfile, dialect='excel', **fmtparams）

　　参数:：

　　　　csvfile：必须是支持迭代(Iterator)的对象，可以是文件(file)对象或者列表(list)对象

　　　　dialect：编码风格，默认为excel的风格默认为excel的风格，也就是用逗号（,）分隔，dialect方式也支持自定义，通过调用register_dialect方法来注册，下文会提到。

　　　　fmtparam：格式化参数，用来覆盖之前dialect对象指定的编码风格。

使用：

import csv

with open('myData.csv', 'r') as csvfile:
    reader_obj = csv.reader(csvfile)
    for line in reader_obj:  # reader_obj对象是一个迭代器
        print(line)

1,a
2,b
3,c
4,d

补充：reader对象还提供一些方法：line_num、dialect、next()

• writer(csvfile, dialect='excel', **fmtparams)

import csv

# 打开文件,newline=''参数,目的是使写入没有多余空行
with open('myData.csv', 'w', newline='') as csvfile:
    writer_obj = csv.writer(csvfile)    # dialect默认excel
    writer_obj.writerow([1,'a'])    # 写入单行
    writer_obj.writerow([2,'b'])
    ls = [(3, 'c'), (4, 'd')]   # 可以写入多行
    writer_obj.writerows(ls)

注意：如果文件't.csv'事先存在，调用writer函数会先清空原文件中的文本，再执行writerow/writerows方法。

补充：除了writerow、writerows，writer对象还提供了其他一些方法：writeheader、dialect

• register_dialect(name, [dialect, ]**fmtparams)