python csv操作

2020-03-25  本文已影响0人  刘年

1、写入列表

#-*-coding:utf-8-*-
#csv的读写
import csv
import codecs #编码,防止出现乱码
#表头,在后续数据分析中,表头很重要,是个重要索引
headers = ['class','name','sex','height','year']
#数据行
rows = [
        [1,'xiaoming','战三',168,23],
        [1,'xiaohong','孩子',162,22],
        [2,'xiaozhang','f全',163,21],
        [2,'xiaoli','十三五',158,21]
    ]
#wb写入不会有空行,但是w每写入一行都会空一行
with open('test.csv','wb')as f:
    '''
    调用writer对象的前提是:
    需要传入一个文件对象,然后才能在这个文件对象的基础上调用csv的写入方法writerow(写入一行)
    writerrow(写入多行)
    '''
    f.write(codecs.BOM_UTF8)
    f_csv = csv.writer(f)
    f_csv.writerow(headers)
    f_csv.writerows(rows)

2、写入字典

#-*-coding:utf-8-*-
#csv的读写
import csv
import codecs #编码,防止出现乱码
headers = ['class','name','sex','height','year']
rows = [
        {'class':1,'name':'小明','sex':'male','height':168,'year':23},
        {'class':1,'name':'王耳朵','sex':'female','height':162,'year':22},
        {'class':2,'name':'xiaozhang','sex':'female','height':163,'year':21},
        {'class':2,'name':'xiaoli','sex':'male','height':158,'year':21},
    ]
with open('test2.csv','wb')as f:
    f.write(codecs.BOM_UTF8)
    f_csv = csv.DictWriter(f,headers)
    f_csv.writeheader()
    f_csv.writerows(rows)

3、读取csv文件

#-*-coding:utf-8-*-
#csv的读写
import csv
with open('test2.csv')as f:
    f_csv = csv.reader(f)
    for row in f_csv:
        print(row[1])
上一篇下一篇

猜你喜欢

热点阅读