selenium自学之路：csv文件读写操作

读取信息

import csv
with open('test.csv')as f:
    f_csv = csv.reader(f)
    for row in f_csv:
        print(row)

运行结果：

['test1', '12345', 'sdasda']
['test2', '44877', 'sdsa']
['test3', '3243', 'fdsgfd']

写入信息：

import csv
headers=['name','age','address']
rows=[
    ['小明',8,'北京'],
    ['小红',7,'天津'],
    ['小刚',8,'武汉']
]

with open('test1.csv','w') as f:
    f_csv = csv.writer(f)
    f_csv.writerow(headers)
    f_csv.writerows(rows)

写入字典数据

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)