码迷,mamicode.com
首页 > 其他好文 > 详细

csv文件读取与写入

时间:2019-07-27 23:58:04      阅读:259      评论:0      收藏:0      [点我收藏+]

标签:创建对象   文件读取   next   head   ade   user   下标   roo   ssr   

csv文件的读取:

1)通过列表下标读取:

import csv
with open(‘stock.csv‘,‘r‘) as fp:
    # reader是个迭代器
    reader = csv.reader(fp)
    next(reader)
    for i in reader:
        # print(i)
        name = i[3]
        volumn = i[-1]
        print({‘name‘:name,‘volumn‘:volumn})

  直接用 open() 函数打开 csv 文件。使用csv.reader() 方法,其中参数为指针。因为该 csv 文件有表头,使用可以使用 next() 函数直接跳过第一组数据,即表头数据。然后直接通过列表下标获取想要的数据。

 

2)通过key获取:

import csv
with open(‘stock.csv‘,‘r‘) as fp:
    reader = csv.DictReader(fp)
    for i in reader:
        value = {"name":i[‘secShortName‘],"volumn":i[‘turnoverVol‘]}
        print(value)

   使用DictReader创建reader对象,不会包含表头那行的数据,而reader这个迭代器与reader创建的又不一样,遍历这个迭代器,返回来的是一个字典,不是列表。

 

csv文件的写入:

1)使用 writer 创建对象,writerow(s) 写入:

import csv
headers = [‘username‘,‘age‘,‘height‘]
value = [
    (‘张三‘,18,180),
    (‘李四‘,19,175),
    (‘王五‘,20,170)
]
with open("classroom.csv",‘w‘,encoding=‘utf-8‘,newline=‘‘) as fp:
    writer = csv.writer(fp)
    writer.writerow(headers)
    writer.writerows(value)

  写入数据到 csv 文件,需要创建一个 writer 对象,才可以使用 writerow 写入一行,而 writerows 是全部写入。其中 默认下 newline=‘\n’ 即写入一行就会换行,所以需要改成空,数据都是存放在列表中。

 

2)使用 DictWriter 创建对象,writerow(s) 写入:

import csv
headers = [‘name‘,‘age‘,‘height‘]
value = [
    {‘name‘:‘张三‘,‘age‘:18,‘height‘:180},
    {‘name‘:‘李四‘,‘age‘:19,‘height‘:175},
    {‘name‘:‘王五‘,‘age‘:20,‘height‘:170}
]
with open("classroom1.csv",‘w‘,encoding=‘utf-8‘,newline=‘‘) as fp: #默认newline=‘\n‘
    writer = csv.DictWriter(fp,headers)
    writer.writeheader()
    writer.writerows(value)

  当数据是存放在字典中可以使用 DictWriter 创建 writer 对象,其中,需要传两个参数,第一个是指针,第二个是表头信息。当使用 DictWriter 创建对象时,写入表头还需要执行 writeheader() 操作。

 

csv文件读取与写入

标签:创建对象   文件读取   next   head   ade   user   下标   roo   ssr   

原文地址:https://www.cnblogs.com/zyde-2893/p/11257211.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!