pythonCSV模块

  在爬虫过后会取得很多信息! 将信息存起来方法还很多中！今天提一下CSV模块

导入模块

import csv

这里先写个列表

rows = [['zhangsan',20],['lisi',22],['wangwu',23],['zhaoliu',18]]

1，写入信息

with open('test.csv','w',newline='')as my_csv:
    write = csv.writer(my_csv)
    for row in rows:
        write.writerow(row)  #写一行

2,读取文件

with open('test.csv','r',newline='')as my_csv:
    reader = csv.reader(my_csv)
    print(type(reader))
    print([row for row in reader])

def write_data():
    colums = int(input('请输入数据的总列数：'))
    my_list =[]
    while 1:
        my_list.append([input('第{}列：'.format(i +1)) for i in range(0,colums)])
        isGoing = input('是否继续循环(Y/N)?')
        if isGoing !='Y':
            break
    print(my_list)

    with open('test2.csv','w',newline='')as my_csv:
        # 通过CSV模块创建一个用于写入数据的对象
        writer = csv.writer(my_csv)
        for row in my_list:
            writer.writerow(row)

3,写入表头数据

list_test = [
    {"name":"python2","version":"2.7"},
    {"name":"python3","version":"2.8"},
    {"name":"python4","version":"2.9"},
    {"name":"python5","version":"3.0"},
    {"name":"python6","version":"3.1"}
]
with open('dict.csv','w',newline='')as my_csv:
    # 创建一个用于写入数据的对象
    keys = [key for key in list_test[0]]
    # print('keys:',keys)
    #fieldnames 用于设置CSV文件的表头内容
    writer = csv.DictWriter(my_csv,fieldnames=keys)
    # 写入表头内容
    writer.writeheader()

    for dict in list_test:
        writer.writerow(dict)

读取上面代码

with open("dict.csv",'r')as my_csv:
    reader = csv.DictReader(my_csv)
    print([d for d in reader])