Python程序中用csv模块来操作csv文件的基本使用教程


Posted in Python onMarch 03, 2016

CSV全称为“Comma Separated Values”,是一种格式化的文件,由行和列组成,分隔符可以根据需要来变化。
如下面为一csv文件:

Title,Release Date,Director
And Now For Something Completely Different,1971,Ian MacNaughton
Monty Python And The Holy Grail,1975,Terry Gilliam and Terry Jones
Monty Python's Life Of Brian,1979,Terry Jones
Monty Python Live At The Hollywood Bowl,1982,Terry Hughes
Monty Python's The Meaning Of Life,1983,Terry Jones

csv可以比较方便的在不同应用之间迁移数据。可以将数据批量导出为csv格式,然后倒入到其他应用程序中。很多应用中需要导出报表,也通常用csv格式导出,然后用Excel工具进行后续编辑。

打印发行日期及标题,逐行处理:

for line in open("samples/sample.csv"):
  title, year, director = line.split(",")
  print year, title

使用csv模块处理:

import csv
reader = csv.reader(open("samples/sample.csv"))
for title, year, director in reader:
  print year, title

改变分隔符

创建一csv.excel的子类,并修改分隔符为”;”

# File: csv-example-2.py
import csv
class SKV(csv.excel):
  # like excel, but uses semicolons
  delimiter = ";"
 
csv.register_dialect("SKV", SKV)
reader = csv.reader(open("samples/sample.skv"), "SKV")
for title, year, director in reader:
  print year, title

如果仅仅仅是改变一两个参数,则可以直接在reader参数中设置,如下:

# File: csv-example-3.py
 
import csv
 
reader = csv.reader(open("samples/sample.skv"), delimiter=";")
 
for title, year, director in reader:
  print year, title

将数据存为CSV格式

通过csv.writer来生成一csv文件。

# File: csv-example-4.py
 
import csv
import sys
 
data = [
  ("And Now For Something Completely Different", 1971, "Ian MacNaughton"),
  ("Monty Python And The Holy Grail", 1975, "Terry Gilliam, Terry Jones"),
  ("Monty Python's Life Of Brian", 1979, "Terry Jones"),
  ("Monty Python Live At The Hollywood Bowl", 1982, "Terry Hughes"),
  ("Monty Python's The Meaning Of Life", 1983, "Terry Jones")
]
 
writer = csv.writer(sys.stdout)
 
for item in data:
  writer.writerow(item)

实例

下面我们来看一个比较完整的例子,代码说明在注释中:

import csv


# dialect是访问csv文件时需要指定的参数之一,用来确定csv文件的数据格式
# 下面这个函数列举系统支持的dialect有哪些,默认值是'excel',用户也可
# 以从Dialect派生一个类,使用该类的实例作为dialect参数。
print csv.list_dialects()


def test_writer():
  # csv文件必须以二进制方式open
  with open('eggs.csv', 'wb') as csvfile:
    spamwriter = csv.writer(csvfile)
    spamwriter.writerow(['Spam'] * 5 + ['Baked Beans'])
    spamwriter.writerow(['Spam', 'Lovely Spam', 'Wonderful Spam'])


def test_reader():
  with open('eggs.csv', 'rb') as csvfile:
    spamreader = csv.reader(csvfile)
    for row in spamreader:
      print row


# sniffer 用来推断csv文件的格式,不是很准确
def test_sniffer():
  with open('eggs.csv', 'wb') as csvfile:
    spamwriter = csv.writer(csvfile, delimiter=' ')

    spamwriter.writerow(['Spam'] * 2 + ['Baked Beans'])
    spamwriter.writerow(['Spam', 'Lovely Spam', 'Wonderful Spam'])

  # 通常你需要指定与写入者相同的文件格式才能正确的读取数据
  with open('eggs.csv', 'rb') as csvfile:
    spamreader = csv.reader(csvfile, delimiter=' ')
    for row in spamreader:
      print ', '.join(row)

  # 如果不知道文件格式,sniffer就可以派上用场了
  with open('eggs.csv', 'rb') as csvfile:
    # 用sniffer推断文件格式,从而得到dialect
    dialect = csv.Sniffer().sniff(csvfile.read(1024))
    print dialect.delimiter, dialect.quotechar

    # 文件重新移动到头部
    csvfile.seek(0)

    # 用推断出来的dialect创建reader
    reader = csv.reader(csvfile, dialect)

    for row in reader:
      print ', '.join(row)
Python 相关文章推荐
Python缩进和冒号详解
Jun 01 Python
解决uWSGI的编码问题详解
Mar 24 Python
python+selenium+autoit实现文件上传功能
Aug 23 Python
python实现跨excel的工作表sheet之间的复制方法
May 03 Python
用Python分析3天破10亿的《我不是药神》到底神在哪?
Jul 12 Python
Python读取txt某几列绘图的方法
Oct 14 Python
python读取txt文件并取其某一列数据的示例
Feb 19 Python
python3实现猜数字游戏
Dec 07 Python
Python  Django 母版和继承解析
Aug 09 Python
pytorch 实现在一个优化器中设置多个网络参数的例子
Feb 20 Python
python 将视频 通过视频帧转换成时间实例
Apr 23 Python
Python迭代器协议及for循环工作机制详解
Jul 14 Python
举例简单讲解Python中的数据存储模块shelve的用法
Mar 03 #Python
Python中的数据对象持久化存储模块pickle的使用示例
Mar 03 #Python
Python和Perl绘制中国北京跑步地图的方法
Mar 03 #Python
python套接字流重定向实例汇总
Mar 03 #Python
Python设计模式中单例模式的实现及在Tornado中的应用
Mar 02 #Python
Python使用设计模式中的责任链模式与迭代器模式的示例
Mar 02 #Python
详解Python设计模式编程中观察者模式与策略模式的运用
Mar 02 #Python
You might like
虚拟主机中对PHP的特殊设置
2006/10/09 PHP
Windows下安装Memcached的步骤说明
2010/04/25 PHP
PHP在不同页面间传递Json数据示例代码
2013/06/08 PHP
PHP+jQuery 注册模块的改进(三):更新到Smarty3.1
2014/10/14 PHP
使用PHP编写发红包程序
2015/07/22 PHP
Laravel实现autoload方法详解
2017/05/07 PHP
PHP实现删除多重数组对象属性并重新赋值的方法
2017/06/07 PHP
Yii2框架类自动加载机制实例分析
2018/05/02 PHP
PHP PDOStatement::fetchAll讲解
2019/01/31 PHP
nginx 设置多个站跨域
2021/03/09 Servers
Javascript中的数学函数集合
2007/05/08 Javascript
鼠标经过的文本框textbox变色
2009/05/21 Javascript
JavaScript Scoping and Hoisting 翻译
2012/07/03 Javascript
javascript面向对象包装类Class封装类库剖析
2013/01/24 Javascript
JS实现点击下载的小例子
2013/07/10 Javascript
使用nodejs、Python写的一个简易HTTP静态文件服务器
2014/07/18 NodeJs
JS实现表单中checkbox对勾选中增加边框显示效果
2015/08/21 Javascript
谈谈Jquery中的children find 的区别有哪些
2015/10/19 Javascript
原生JS轮播图插件
2017/02/09 Javascript
vue自定义指令directive的使用方法
2019/04/07 Javascript
Vue实现input宽度随文字长度自适应操作
2020/07/29 Javascript
vue+高德地图实现地图搜索及点击定位操作
2020/09/09 Javascript
[08:38]DOTA2-DPC中国联赛 正赛 VG vs Elephant 选手采访
2021/03/11 DOTA
python进阶教程之异常处理
2014/08/30 Python
python 3.0 模拟用户登录功能并实现三次错误锁定
2017/11/01 Python
Python实现的连接mssql数据库操作示例
2018/08/17 Python
python 在某.py文件中调用其他.py内的函数的方法
2019/06/25 Python
python飞机大战pygame碰撞检测实现方法分析
2019/12/17 Python
python利用platform模块获取系统信息
2020/10/09 Python
中专生毕业自我鉴定
2013/11/01 职场文书
综治工作心得体会
2014/09/11 职场文书
交通安全横幅标语
2014/10/07 职场文书
雷峰塔导游词
2015/02/09 职场文书
刑事附带民事诉讼答辩状
2015/05/22 职场文书
初中军训感言
2015/08/01 职场文书
《一面五星红旗》教学反思
2016/02/23 职场文书