Python3:csv的读写

发布时间:2019-09-28 08:38:33编辑:auto阅读(2381)

    版权声明:本文为CSDN博主「katyusha1」的原创文章,遵循CC 4.0 by-sa版权协议,转载请附上原文出处链接及本声明。

    原文链接:https://blog.csdn.net/katyusha1/article/details/81606175


    前言

    快要毕业那会儿,在下编写了一个招聘网站招聘岗位的爬虫提供给前女神参考,最开始我是存到mysql中,然后在到处一份csv文件给前女神。到了参加工作后,由于经常使用excel绘制图表(谁叫公司做报表全用excel呢…………….),在下才瞬间顿悟,有时候将爬虫爬取的数据直接存到csv中或许会更加方便一些!


    既然涉及到爬虫的写入文件,那这里我就先介绍python对csv的写!

    1、csv的写

    python自带了csv模块提供用户对csv文件进行读写操作,要对csv文件进行写操作,首先要创建一个writer对象,参考help(csv.writer),情况如下 

     

    调用writer对象的前提是:需要传入一个文件对象,然后才能在这个文件对象的基础上调用csv的写入方法writerow(写入一行)writerrow(写入多行)。写入数据的代码如下:


    import csv


    headers = ['class','name','sex','height','year']


    rows = [

            [1,'xiaoming','male',168,23],

            [1,'xiaohong','female',162,22],

            [2,'xiaozhang','female',163,21],

            [2,'xiaoli','male',158,21]

        ]


    with open('test.csv','w')as f:

        f_csv = csv.writer(f)

        f_csv.writerow(headers)

        f_csv.writerows(rows)

    1

    2

    3

    4

    5

    6

    7

    8

    9

    10

    11

    12

    13

    14

    15

    这段代码我首先定义了写入csv文件的表头、每一列的内容,然后打开一个csv文件,将文件对象作为参数传给csv.writer(),最后将表头和每一行的内容写入到csv文件中。 

    执行结果: 



    注意:如果打开csv文件出现空行的情况,那么需要添加一个参数 newline=”(我使用windows出现了这种情况,使用linux mint没有出现)

    with open('test.csv','w',newline='')as f:

    1

    2、写入字典序列的数据

     

    参考help(csv.DictWriter)可知,在写入字典序列类型数据的时候,需要传入两个参数,一个是文件对象——f,一个是字段名称——fieldnames,到时候要写入表头的时候,只需要调用writerheader方法,写入一行字典系列数据调用writerrow方法,并传入相应字典参数,写入多行调用writerows 

    具体代码如下:


    import csv


    headers = ['class','name','sex','height','year']


    rows = [

            {'class':1,'name':'xiaoming','sex':'male','height':168,'year':23},

            {'class':1,'name':'xiaohong','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','w',newline='')as f:

        f_csv = csv.DictWriter(f,headers)

        f_csv.writeheader()

        f_csv.writerows(rows)

    1

    2

    3

    4

    5

    6

    7

    8

    9

    10

    11

    12

    13

    14

    15

    运行结果: 



    3、csv文件的读

    读取csv时需要使用reader,并传如一个文件对象,而且reader返回的是一个可迭代的对象,需要使用for循环遍历,代码如下:


    import csv

    with open('test.csv')as f:

        f_csv = csv.reader(f)

        for row in f_csv:

            print(row)

    1

    2

    3

    4

    5

    运行结果: 

     

    在上面,row是一个列表,如果想要查看固定的某列,则需要加上下标,例如我想要查看name,那么只需要改为row[1]


    import csv

    with open('test.csv')as f:

        f_csv = csv.reader(f)

        for row in f_csv:

            print(row[1])

    1

    2

    3

    4

    5


    --------------------- 

    版权声明:本文为CSDN博主「katyusha1」的原创文章,遵循CC 4.0 by-sa版权协议,转载请附上原文出处链接及本声明。

    原文链接:https://blog.csdn.net/katyusha1/article/details/81606175前言

    快要毕业那会儿,在下编写了一个招聘网站招聘岗位的爬虫提供给前女神参考,最开始我是存到mysql中,然后在到处一份csv文件给前女神。到了参加工作后,由于经常使用excel绘制图表(谁叫公司做报表全用excel呢…………….),在下才瞬间顿悟,有时候将爬虫爬取的数据直接存到csv中或许会更加方便一些!


    既然涉及到爬虫的写入文件,那这里我就先介绍python对csv的写!

    1、csv的写

    python自带了csv模块提供用户对csv文件进行读写操作,要对csv文件进行写操作,首先要创建一个writer对象,参考help(csv.writer),情况如下 

     

    调用writer对象的前提是:需要传入一个文件对象,然后才能在这个文件对象的基础上调用csv的写入方法writerow(写入一行)writerrow(写入多行)。写入数据的代码如下:


    import csv


    headers = ['class','name','sex','height','year']


    rows = [

            [1,'xiaoming','male',168,23],

            [1,'xiaohong','female',162,22],

            [2,'xiaozhang','female',163,21],

            [2,'xiaoli','male',158,21]

        ]


    with open('test.csv','w')as f:

        f_csv = csv.writer(f)

        f_csv.writerow(headers)

        f_csv.writerows(rows)

    1

    2

    3

    4

    5

    6

    7

    8

    9

    10

    11

    12

    13

    14

    15

    这段代码我首先定义了写入csv文件的表头、每一列的内容,然后打开一个csv文件,将文件对象作为参数传给csv.writer(),最后将表头和每一行的内容写入到csv文件中。 

    执行结果: 



    注意:如果打开csv文件出现空行的情况,那么需要添加一个参数 newline=”(我使用windows出现了这种情况,使用linux mint没有出现)

    with open('test.csv','w',newline='')as f:

    1

    2、写入字典序列的数据

     

    参考help(csv.DictWriter)可知,在写入字典序列类型数据的时候,需要传入两个参数,一个是文件对象——f,一个是字段名称——fieldnames,到时候要写入表头的时候,只需要调用writerheader方法,写入一行字典系列数据调用writerrow方法,并传入相应字典参数,写入多行调用writerows 

    具体代码如下:


    import csv


    headers = ['class','name','sex','height','year']


    rows = [

            {'class':1,'name':'xiaoming','sex':'male','height':168,'year':23},

            {'class':1,'name':'xiaohong','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','w',newline='')as f:

        f_csv = csv.DictWriter(f,headers)

        f_csv.writeheader()

        f_csv.writerows(rows)

    1

    2

    3

    4

    5

    6

    7

    8

    9

    10

    11

    12

    13

    14

    15

    运行结果: 



    3、csv文件的读

    读取csv时需要使用reader,并传如一个文件对象,而且reader返回的是一个可迭代的对象,需要使用for循环遍历,代码如下:


    import csv

    with open('test.csv')as f:

        f_csv = csv.reader(f)

        for row in f_csv:

            print(row)

    1

    2

    3

    4

    5

    运行结果: 

     

    在上面,row是一个列表,如果想要查看固定的某列,则需要加上下标,例如我想要查看name,那么只需要改为row[1]


    import csv

    with open('test.csv')as f:

        f_csv = csv.reader(f)

        for row in f_csv:

            print(row[1])

    1

    2

    3

    4

    5



关键字

上一篇: python3操作串口

下一篇: Python3 元组