Python CSV模块使用实例

Python 的 CSV模块的使用方法，包括，reader, writer, DictReader, DictWriter.register_dialect

读取CSV文件
reader(csvfile[, dialect=‘excel’][, fmtparam])
参数分别为：1、要打开的csv文件。2、编码风格，默认为excel方式。3、读写模式

import csv
csvFile = open('my.csv', 'rb')
reader = csv.reader(csvFile)
for line in reader:
    print line
csvFile.close()

with open("my.csv",'r') as csvFile:
    #读取csv文件,返回的是迭代类型
    reader2 = csv.reader(csvFile)
    for item in reader2:
        print(item)

读写csv文件实例

#第一种方法:使用reader函数,接收一个可迭代的对象(比如csv文件),能返回一个生成器,就可以从其中解析出csv的内容.eg:下面的代码可以读取csv的全部内容,以行为单位
with open('test.csv','rb') as csvFile:
    reader = csv.reader(csvFile)
    rows = [row for row in reader]
  
print('===========')
  
#提取其中某一列可以使用下面的方法
with open('test.csv','rb') as csvFile:
    reader = csv.reader(csvFile)
    column = [row[1] for row in reader]
  
  
print('======第二种方法======')
#第二种方法:使用DictReader,和reader函数类似,接收可迭代的对象,能返回一个可迭代的对象,能返回一个生成器,但是返回的每一个单元格都放在一个字典的值内,而这个字典的键则是这个单元格的标题.用下面的代码可以看到DictReader的结构:
with open('test.csv','rb') as csvFile:
    reader = csv.DictReader(csvFile)
    column = [row for row in reader]
print(column)
  
#使用CictReader读取csv的某一列,这样可以按照列的标题查询
with open('test.csv','rb') as csvFile:
    reader = csv.DictReader(csvFile)
    column = [row['a_name'] for row in reader]

写入数据

with open('data3.csv','w',newline='') as csvfile:
        f_csv = csv.writer(csvfile)
        f_csv.writerows(data)   #writerow单行写入，writerows多行写入